实验室经常需要大家共用一个服务器,因此不同实验者有不同的cuda需求,况且同一个人在不同项目下也有不同的cuda需求,因此需要更换cuda。

查询当前版本

首先通过下面命令查询当前的cuda版本:

nvcc -V

得到如下结果

代表当前的版本是11.8

查询可更换版本

因为安装cuda是需要root的,如果自身没有root,则需要查看有哪些cuda可用。一般实验机器都有安装多个版本。通过这个命令来查询可更换版本。

ls -l /usr/local | grep cuda

得到结果如下:

这就说明有一系列的11和12cuda可用

更换相应的cuda版本

有两种更换模式,例子更换11.7

  1. 暂时更换
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会得到:

更换成功

  1. 永久更换(谨慎)

通过写入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  # 使修改生效
最后修改:2025 年 03 月 20 日
如果觉得我的文章对你有用,请随意赞赏