做到真正的一行代码搞定,只需要修改对应软链接,就可以轻松实现快捷切换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上的共存与管理