在已存在多种版本cuda的情况下,切换cuda版本以应对不同的pytorch版本

usr/loacl目录中可以查看存在的cuda版本,可以看到我的服务器上包含了10.0、10.1、10.2和11.1四种版本的cuda(没有的需要自行下载)
在这里插入图片描述
然后在主目录下运行:

vim .bashrc

看看是不是空的,不是空的表明打开的是对的,然后在最下面加上

export CUDA_HOME=/usr/local/cuda-11.1
export PATH=/usr/local/cuda-11.1/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

这里的路径与usr/loacl中的对应,然后

source .bashrc

就可以了,然后通过nvcc -V查看当前cuda版本,可以看到已经成功切换了,然后就可以开始安装更高版本的pytorch了。
在这里插入图片描述

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐