实验室经常需要大家共用一个服务器,因此不同实验者有不同的cuda需求,况且同一个人在不同项目下也有不同的cuda需求,因此需要更换cuda。
查询当前版本
首先通过下面命令查询当前的cuda版本:
nvcc -V得到如下结果

代表当前的版本是11.8
查询可更换版本
因为安装cuda是需要root的,如果自身没有root,则需要查看有哪些cuda可用。一般实验机器都有安装多个版本。通过这个命令来查询可更换版本。
ls -l /usr/local | grep cuda得到结果如下:
这就说明有一系列的11和12cuda可用
更换相应的cuda版本
有两种更换模式,例子更换11.7
- 暂时更换
export CUDA_HOME=/usr/local/cuda-11.7
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH输入后再nvcc -V会得到:

更换成功
- 永久更换(谨慎)
通过写入bashrc文件可以永久更换,也就是说每次终端开始的时候都会更换,需要确认是否有这个需求:
echo "export CUDA_HOME=/usr/local/cuda-11.7" >> ~/.bashrc
echo "export PATH=$CUDA_HOME/bin:$PATH" >> ~/.bashrc
echo "export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH" >> ~/.bashrc然后运行:
source ~/.bashrc # 使修改生效
1 条评论
[...]在这篇文章当中,介绍了如何在机器上查看可以使用的cuda版本,同时有暂时更换cuda的方法以及永久更换cuda的方法。但是无论是哪种,都需要手动输入命令。于是我让ai写了一个脚本,能够通过运行脚本快速地更换cuda版本:首先还是需要查看机器能够使用的cuda有哪些:nvcc -V然后脚本内容如下:# Accept version as parameter, default to 111 if no[...]