ubuntu多cuda版本切换

ctrl+alt+t进入终端

>cd /usr/local
>ls
ubuntu多cuda版本切换_第1张图片
图片.png

这个ubuntu装了两个cuda分别是cuda10.0和cuda10.1

 >stat cuda

ubuntu多cuda版本切换_第2张图片
图片.png

当前软链接指向cuda10.0

当我们想使用cuda-10.1版本时,只需要删除cuda10.0软链接,然后重新建立指向cuda-10.1版本的软链接即可(注意名称还是cuda,因为要与bashrc文件里设置的保持一致)

    sudo rm -rf cuda
    sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda

想切换其他版本的cuda,只需要改动建立软链接时cdua的正确路径即可

但是当nvcc -V时,还是显示cuda10.0

图片.png

所以我们查看环境变量path,命令:
echo $PATH

图片.png

发现原因:是因为/usr/bin/cuda-10.0(①处)在PATH中相较/usr/local/cuda-10.1(②处)靠前,所以nvcc –V命令会按照PATH顺序,先去①里面找,应该是在这文件夹里的nvcc中找到了10.0版本的cuda,所以就不去别的地方找了,就没有找到②处我们想要的cuda版本;

更改PATH内①和②顺序,采用的方法是将环境变量②位置提前:

export PATH=/usr/local/cuda/bin:$PATH

nvcc -V

图片.png

显示CUDA10.1了

显示CUDNN版本7.6.5

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
ubuntu多cuda版本切换_第3张图片
图片.png

切换CUDA软链接,cudnn版本也会相应改变。

cuda10.1------>cudnn7.6.5
cuda10.0------>cudnn7.4.2

你可能感兴趣的:(ubuntu多cuda版本切换)