原因
由于服务器中安装了两个版本的cuda(cuda10.1和cuda11.1),不同项目可能需要应用不同的cuda版本,但是自己又没有root权限或者只想在使用指定conda环境时改为用指定的cuda版本。总结起来有三种方法:
1、修改软链接指向,但是这样会使得所有环境以及所有账户的cuda版本都改变,会影响其他人**(强烈不推荐)**
2、修改 ~.bashrc 文件(修改方法见“不同linux账户切换不同的cuda版本”),这样不会改变其他账户的cuda版本,但是自己账户下的所有conda环境都会改变cuda版本,有时候不同conda环境安装的torch版本不一样,用到的cuda版本也可能不一样。(不太推荐)
3、增加env_vars.sh 文件 (修改方法见“不同conda环境切换不同的cuda版本”)。这样修改则是不同的conda环境使用的cuda版本不一样(强烈推荐)。
下面介绍方法2的修改
修改步骤
1、查看默认的cuda版本
一般而言,默认会在/usr/local/下面创建一个软链接,指向某个cuda版本。
如下图
cd /usr/local 定位到目标夹
ls -al 即可查看,cuda指向的是10.1版本
我们来验证一下是不