Linux下CUDA的多版本安装及切换

Linux下CUDA的多版本安装及切换

一.安装CUDA

1.下载CUDA安装包(如：cuda_10.0.130_410.48_linux.run)
2.cd到cuda_10.0.130_410.48_linux.run文件所在的文件夹下执行：

sudo ./cuda_10.0.130_410.48_linux.run

如果执行不了，添加可执行权限后再执行：

sudo chmod a+x cuda_10.0.130_410.48_linux.run
sudo ./cuda_10.0.130_410.48_linux.run

按下面的选项选择：
3.添加环境变量

sudo gedit ~/.bashrc

添加下面两句

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

执行：

source ~/.bashrc

二.安装cudnn

1.下载与CUDA对应的cudnn版本
2.cd到cudnn所在的文件夹下进行解压：

tar -zxvf cudnn-10.0-linux-x64-v7.6.5.32.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ 
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ 
sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h /usr/local/cuda-10.0/lib64/libcudnn*
     
cd /usr/local/cuda-10.0/targets/x86_64-linux/lib/
sudo ln -sf libcudnn.so.7.6.5 libcudnn.so.7

三.CUDA版本间的切换

安装了多个CUDA的版本，那么可以在这些CUDA的版本之间切换(切换后记得安装对应的cudnn的版本)
在安装了多个cuda版本后，可以在/usr/local/目录下查看自己安装的cuda版本
只需要删除cuda软链接，然后重新建立指向另一个版本的软链接即可

sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda

四.cudnn版本切换

删除原来的版本：

sudo rm -rf /usr/local/cuda-10.2/include/cudnn.h
sudo rm -rf /usr/local/cuda-10.2/lib64/libcudnn*

添加新版本：

sudo cp include/cudnn.h /usr/local/cuda-10.2/include/
sudo cp lib64/lib* /usr/local/cuda-10.2/lib64/
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h /usr/local/cuda-10.2/lib64/libcudnn*
     
cd /usr/local/cuda-10.2/targets/x86_64-linux/lib/
sudo ln -sf libcudnn.so.7.6.5 libcudnn.so.7