最快配置GPU版本Pytorch

GPU版Pytorch

最近在搞深度学习，需要用到GPU跑模型了，配置GPU版本的Pytorch是真的烦，而且每个人遇到的问题不一样，网上很多教程也良莠不齐，这里我将自己总结的最简便的配置Pytorch方法分享给大家。

1.下载安装CUDA

首先需要查看当前电脑显卡CUDA版本：

右键点击图标进入控制面板–>帮助–>系统信息–>组件，出现如下界面，比如我的就是CUDA 10.1版本：

CUDA官网地址：https://developer.nvidia.com/cuda-toolkit-archive
然后去官网找到对应CUDA，比如我的是10.1，点击进去下载安装即可
注意：一定要对应自己电脑的版本


下载完成后是一个exe文件，双击进行安装：

安装过程不用多说，没什么问题，安装好了一定要记得自己安装的位置，一般都是安装在C盘里面。
安装好了以后可以通过打开命令行，输入：

nvcc -V

如果出现以上反应，就代表安装好了，不然就重新安装。

2.下载安装cuDNN

cuDNN下载地址：https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b
同理找到对应自己CUDA版本的cuDNN然后下载就行了，比如我的是10.1，就下载一个对应版本的：

下载完毕后是一个压缩包，将压缩包进行解压，解压后是一个名为cuda的文件夹：

解压后：

将解压后的文件夹cuda改为cudnn，然后复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1里面：

然后进行最后一步，配置环境变量：
将如下两个路径写入系统变量的Path下面：


然后其他的就可以不用管了。大功告成！

3.最后一步，安装Pytorch和torchvision

网上清一色都是去Pytorch官网下载然后用命令行安装，但是我用那种方法没有成功，而且就算成功了也非常慢，因为要下载安装很久，还有意想不到的报错，所以我用了最快也最稳妥的方法，就是去这个网站上面下载torch和torchvision，然后直接pip安装就完事了，下载非常快，还绝对不会报错。
网站入口：https://download.pytorch.org/whl/torch_stable.html

进入网站，发现非常多版本，怎么找呢？
注意：前缀CU代表GPU版本，101就代表10.1，然后cp37代表python3.7版本，这几个参数可能不太明白需要解释，然后你只需要下载自己电脑CUDA对应的torch版本就行了，然后根据下图torch和torchvision的对应关系找torchvision的版本就行了：

下载完成后，找到对应下载文件夹，然后打开命令行进行pip命令安装torch和torchvision：
例如：

pip install .\torchvision-0.6.1+cu101-cp37-cp37m-win_amd64.whl

安装完成后，在命令行直接进入python环境，运行如下代码：

import torch
torch.cuda.is_available()

如果像我上面一样运行出来结果是True，那证明安装成功了，如果是False或者报错就是没有安装成功，那可能就是你的版本对应关系出问题了！

写在最后

我这种方法是最快也最稳妥的方法，本人才疏学浅，如果有误请指教！