Ubuntu20.04下安装多CUDA版本，以及后续切换卸载
本方案的前提是假设机子上已经有一个版本的cuda，现在需要支持新的torch2.1.2和torchvision==0.16.2，于是来安装新的cuda
一、选择版本
如果我想安装支持torch2.1.2的cuda版本，到官网（https://pytorch.org/get-started/previous-versions/#v200）查看版本对应，发现11.8或者12.1可以支持torch2.1.2和torchvision==0.16.2
nvidia-smi
查看自己的电脑能支持什么版本，根据结果我们的设备最高只能运行12.0版本，达不到12.1，所以选择安装11.8
二、安装CUDA
到官网找到11.8：https://developer.nvidia.com/cuda-toolkit-archive，根据指引进行安装，即运行以下命令
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run
sudo sh cuda_11.8.0_520.61.05_linux.run
运行run之后，取消选中Driver，其他默认，然后Install。
【注意如果这里没有Driver，可以参考其他博客先安装好Nvidia显卡驱动】
三、安装Cudnn
进入官网：https://developer.nvidia.com/rdp/cudnn-archive，我要安装CUDA11.8，所以选择for CUDA 11.x下的内容
下载完成后，运行以下命令来解压和配置
# 解压
tar -vxf ./cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz
# 复制
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda11-archive/include/* /usr/local/cuda-11.8/include
sudo cp cudnn-linux-x86_64-8.9.7.29_cuda11-archive/lib/libcudnn* /usr/local/cuda-11.8/lib64
# 添加读写权限
sudo chmod a+r /usr/local/cuda-11.8/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.8/lib64/libcudnn*
cat /usr/local/cuda-11.8/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 # 测试安装是否正常
最后如下图显示则说明安装成功
四、环境配置
最后运行官网中一开始的命令来完成安装（没有环境的先用前两行代码创建一个）
# conda create -n torch21 python=3.10 -y
# conda activate torch
conda install pytorch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 pytorch-cuda=11.8 -c pytorch -c nvidia
五、其他操作（切换/卸载CUDA版本）
（1）切换cuda版本
安装的时候我们电脑下可能已经有默认的cuda版本，安装新版本是是会支持覆盖的
另外也可以通过以下命令切换
sudo rm -rf /usr/local/cuda #删除之前创建的软链接
sudo ln -s /usr/local/cuda-11.8 /usr/local/cuda #创建链接到cuda-11.0的软链接
nvcc -V #查看当前cuda版本
（2）卸载CUDA
如果想卸载CUDA，进入对应路径，运行以下命令后选中相应的（以卸载CUDA12.0为例），然后Done。
cd /usr/local/cuda-12.0/bin
sudo ./cuda-uninstaller
参考链接：
ubuntu配置多版本cuda+cudnn环境，及版本切换方法
原文地址：https://blog.csdn.net/Netceor/article/details/142921173
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！