Linux系统中,多个CUDA版本切换

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

设置完毕后如下图所示:Linux系统中,多个CUDA版本切换_第1张图片
**注意:**如果大家按照下图设置的话,这是硬链接,无法设置软连接。也就是说,即使你按照步骤2设置完软连接,输入nvcc -V,发现没能切换cuda版本
Linux系统中,多个CUDA版本切换_第2张图片
输入下面的指令,使修改后的bashrc生效

source ~/.bashrc

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

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

cd /usr/local
stat cuda

Linux系统中,多个CUDA版本切换_第3张图片
通过输入指令nvcc -V,查看当前的runtime cuda版本是11.0
Linux系统中,多个CUDA版本切换_第4张图片
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版本切换_第5张图片

你可能感兴趣的:(工具使用,python,pycharm,linux,cuda)