Ubuntu18.04任意切换多个版本的CUDA与CUDNN
CUDA 的下载与安装
1、CUDA下载与安装
各版本CUDA下载地址 https://developer.nvidia.com/cuda-toolkit-archive 选择需要的版本后,下载离线安装包。 安装刚刚下载的文件 输入accept接受 如果你已经安装Nvidia显卡驱动,取消驱动安装,只安装cuda。 如果弹出
Do you want to install a synbolic link at usr/local/cuda?
如果你是第一次安装cuda,选择yes创建软链接,否则选择no。
我这里安装了两个版本的cuda
CUDA 版本切换
将对应版本的CUDA路径添加到环境
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.2/lib64
export PATH=$PATH:/usr/local/cuda-11.2/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.2
更新环境(后面想用另外一个版本的cuda,注释另外一个版本,重新更新环境即可)
source ~/.bashrc
使用命令查看版本
nvcc -V
2、cudnn安装
各版本cudnn下载地址 https://developer.nvidia.com/rdp/cudnn-archive 下载之前请确认cuda对应的cudnn版本
解压下载的文件,移动至对应的cuda目录并修改权限。
sudo cp cudnn/include/cudnn*.h /usr/local/cuda/include
sudo cp cudnn/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
至此大功告成,如有错误还请指出!!
|