ubuntu多cuda版本切换

ctrl+alt+t进入终端

>cd /usr/local
>ls
图片.png

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

 >stat cuda

图片.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
图片.png

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

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

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容