Ubuntu切换不同版本的cuda

切换cuda版本

Ubuntu系统往往需要切换不同版本的cuda，本教程提供了一种便捷的cuda版本切换方式。

首先打开/usr/local/文件夹，查看安装的cuda版本

如上图，系统里有cuda10.1和cuda10.0两个版本。

然后查看当前cuda对应版本：

nvcc -V
1

打开bashrc文件夹：

gedit ~/.bashrc
1

添加以下内容：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
1
2
3

终端输入：

source ~/.bashrc
1

建立新版本cuda软链接

sudo rm -rf /usr/local/cuda  #删除之前生成的软链接
sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda #生成新的软链接
1
2

检查是否建立新版本的cuda

nvcc -V
1

切换cudnn版本

• 下载cudnn：
  https://developer.nvidia.com/rdp/cudnn-archive
  在主文件夹解压。
• 在cuda文件夹所在的路径打开终端，在这个路径下依次执行下面四个命令，将cudnn的文件复制到cuda的安装环境中：

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/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
1
2
3
4

• 添加软链接

cd /usr/local/cuda/lib64
sudo rm -r libcudnn.so libcudnn.so.7  #删除之前生成的软链接
sudo ln -s libcudnn.so.7.6.5 libcudnn.so.7 #生成新的软链接
sudo ln -s libcudnn.so.7 libcudnn.so #生成新的软链接
1
2
3
4