首先说明一点,我们的Ubuntu20.04系统上,可以安装多个版本的CUDA,如下图所示,cuda11.0、cuda11.1和cuda11.3
进入目录 cd /usr/local/
显示当前目录 ls -alh
并且,在安装多个版本CUDA时,会有提示如下:
这个提示:是在/usr/local/cuda路径下已经存在一个名为cuda的符号链接,安装程序检测到这一点并询问用户是否要更新/覆盖该符号链接,从而保证新的安装程序正确地安装在该路径下。
这就给我们提供了思路,这个CUDA符号链接可以修改。
配置环境变量的cuda不要指定具体的版本,是为了电脑中多个CUDA版本的切换,环境变量的地址为 /usr/local/cuda,
vi ~/.bashrc文件查看设置如下:
如何进行多版本的切换呢,比如想把CUDA版本切换成11.3版本(但是要保留CUDA 11.1和11.0版本),我们只需要修改软链接即可,将CUDA 11.3的软链接链接到cuda目录下,代码如下:
sudo rm -rf cuda # 删除原有的软链接
sudo ln -s /usr/local/cuda-11.3 /usr/local/cuda # 将cuda-11.3链接到cuda下
软链接重新生成后,使用 nvcc -V 命令可以查看当前的CUDA版本,如下所示: