开始时,我需要安装jax-0.4.20版本,但该版本需要CUDA-12.3和cudnn8.9的配套包。在服务器上,我通过nvidia-smi命令查看到Driver Version为470.57.02,这个驱动的兼容CUDA Version上限为11.4。
查阅英伟达官网关于CUDA兼容性的信息后,我发现470.57.02驱动支持向后兼容,这意味着即便驱动版本较低,仍可安装较高版本的CUDA runtime。与以往不同,旧驱动不支持此特性,例如R418的驱动,升级到R450是安装CUDA 10.1到11.0的必要条件。
从R470驱动版本开始,CUDA兼容性变得可用。安装时,仅需用户模式库和所需的CUDA版本,如CUDA-11.0。然而,并非所有驱动如R525、R495等都支持此兼容性。
举两个实例:CUDA-12.4在R470和R535驱动下兼容,但R550无需额外安装;在470驱动上安装CUDA-12.3,可以使用兼容包进行安装。关于安装步骤,可以参考官方指南。
在线安装兼容包,首先在最高支持CUDA-11.4的470.57.02上安装CUDA-12.3,通常下载到/usr/local/cuda-12.3目录。创建一个compat子目录存放兼容包文件,这些文件是新版本驱动中的几个库,需要下载对应驱动,例如NVIDIA-Linux-x86_64-550.54.15.run,然后解压并复制特定库到compat目录,通过软链接建立指向。
最后,添加环境变量`LD_LIBRARY_PATH`,如`/usr/local/cuda/compat:$LD_LIBRARY_PATH`,以确保库路径正确。安装成功后,可以使用deviceQuery进行检查,具体步骤可以参考相关教程链接。
本文如未解决您的问题请添加抖音号:51dongshi(抖音搜索懂视),直接咨询即可。