Ubuntu20.04快速切换多个版本的Pytorch和CUDA

一. 安装CUDA-10.2

首先安装驱动,安装驱动参考上一篇.
接下来进行CUDA-10.2的安装,使用以下命令可以忽略已经安装驱动的问题:

sudo sh cuda_10.2.89_440.33.01_linux.run --override

安装成功:

二. 配置.bashrc文件

打开.bashrc文件:

sudo gedit ~/.bashrc

把这里原来cuda-11.0改成共用的cuda:

改为:

export PATH="$PATH:/usr/local/cuda/bin"
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64/"
export LIBRARY_PATH="$LIBRARY_PATH:/usr/local/cuda/lib64"  

使其生效:

source  ~/.bashrc

三. 设置软链接

1.首先查看已有的软链接:

stat /usr/local/cuda

发现是cuda-11.1链接到cuda了,下面改成cuda-10.2,首先删除原有的软链接:

sudo rm -rf /usr/local/cuda

后面在用的时候直接设置软链接:

sudo ln -s /usr/local/cuda-10.2 /usr/local/cuda

发现链接成功: