切换cuda版本
在已存在多种版本的cuda情况下,切换cuda版本以应对更高版本的pytorch。
·
在已存在多种版本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了。
欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
已为社区贡献1条内容
所有评论(0)