admin 管理员组

文章数量: 1184232

1. 设置环境变量

sudo vi ~/.bashrc

进入bashrc文件后,在文件末尾添加以下代码,用来设置cuda的环境变量:

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

设置完毕后如下图所示:
**注意:**如果大家按照下图设置的话,这是硬链接,无法设置软连接。也就是说,即使你按照步骤2设置完软连接,输入nvcc -V,发现没能切换cuda版本

输入下面的指令,使修改后的bashrc生效

source ~/.bashrc

2. 通过设置软连接,切换cuda版本

查看当前软连接的cuda版本,是cuda-11.0

cd /usr/local
stat cuda


通过输入指令nvcc -V,查看当前的runtime cuda版本是11.0

cuda-11.0切换成cuda-10.2

sudo rm -rf /usr/local/cuda
sudo ln -s /usr/local/cuda-10.0 /usr/local/cuda
stat cuda
nvcc -V

本文标签: 多个 版本 系统 Linux CUDA