Linux多版本cuda切换

目标

将cuda版本从10.0切换为11.1

步骤

查看当前cuda版本：

nvcc -V

编辑.bashrc文件：

vim ~/.bashrc

在文件中添加以下几行（若已存在则忽略）：

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

保存关闭后，运行：

source ~/.bashrc

进入cuda所在文件夹：

cd /usr/local

可以看到当前目录下安装了两个版本的cuda，分别为cuda-10.0和cuda-11.1

删除原来的软链接：

sudo rm -rf cuda

添加软链接：

sudo ln -s cuda-11.1 cuda

查看cuda版本验证是否切换成功：

nvcc -V

若没有变化，可能需要关闭终端重新打开