[清爽快捷] Ubuntu上多个版本的cuda切换

做到真正的一行代码搞定，只需要修改对应软链接，就可以轻松实现快捷切换cuda

查看已安装的cuda版本有哪些

一般如果我们都是使用默认位置安装cuda的话，那么其安装路径都是/usr/local。如果要查看该目录下已经安装有哪些版本的cuda，可以通过在终端执行如下命令：

ls -l /usr/local | grep cuda 

参考：ubuntu下安装多版本cuda及版本切换教程_ubuntu切换cuda版本-CSDN博客

如果没有自己想要的版本，那请先安装cuda安装教程。不需要删除之前版本的。

检查环境变量

由于我们建立了软链接，可以直接将相关的环境变量设置成含有软链接cuda的，这样就能减少切换CUDA版本的工作量。
使用vim ~/.bashrc进入环境变量的编辑模式之后，根据实际情况，新增或者设置下面三项。
如果使用的是zsh，那么vim ~/.zshrc

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

设置软连接方便切换(重点)

• CUDA一般的安装目录在/usr/local里面，因此进入该目录后，会发现这个目录里面已经存在两个或以上不同版本的CUDA，判断的方法就是文件夹的名称——一般都会采用cuda+版本号来命名这些文件夹。如果目录里面已经存在cuda文件夹，则说明已经存在一个软链接，链接到特定版本的CUDA了。可以通过stat命令查看这些关系
• 然后就可以使用ls命令修改软连接

stat /usr/local/cuda #查看当前cuda的软连接指向
sudo rm -rf /usr/local/cuda #如果有误，则删除
sudo ln -s /usr/local/cuda-11.8 /usr/local/cuda # 创建我们指定的cuda版本的软连接

参考：多个版本的CUDA在Ubuntu上的共存与管理