解决Pytorch的GPU版本安装问题

解决Pytorch的GPU版本安装问题

在安装Pytorch GPU版本的时候，我们常常遇到GPU不支持的情况，这个过程说明我们的Pytorch安装过程出现了问题，以下是解决方法

查看自己的CUDA显卡驱动版本

在安装Pytorch前， 我们需要判断自己的显卡驱动版本

在cmd命令行中输入
nvidia-smi

可以看到我们的CUDA Version对应的版本是12.7

打开Pytorch官网(https://pytorch.org/)国外网站，加载速度可能较慢

方法1：直接安装

我的CUDA对应版本是12.7，但是可供选择的没有12.7版本，在这里我推荐使用11.8的版本，是最稳定的版本

方法2：手动安装

直接安装的方法可能会遇到下载速度慢的问题，这里我推荐使用手动离线安装的方法，下载速度较快

离线安装地址：（https://pytorch.org/get-started/previous-versions/）

下划我们找到一个版本，这里一定要记住torch、torchvision、torchaudio三个包对应的版本，然后我们找到对应CUDA 11.8的网址

找到这几个包，点进去安装对应的版本（上图找到的对应版本）

torch这里的cu118表示CUDA11.8 cp310表示Python 3.10版本 win是windows版本

点击安装，推荐使用迅雷，下载比较快

torchaudio同理

torchvision 同理找到对应的版本

下载完成之后，定位到包下载的目录，最后依次输入 pip install 加上包名

最后验证安装

print(torch.cuda.is_available()) #输出True则验证安装成功