本方案的前提是假设机子上已经有一个版本的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,根据指引进行安装,即运行以下命令