切换cuda版本

在已存在多种版本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了。