安装GPU版PyTorch

点击之后会出现如下图，找到自己电脑对应的版本安装，在右下角点击下载，所占内存有点大的。

下载好后安装在自己想安装的位置。（！这里我安装的时候出现了一点小事故，我原本在C盘新建了一个文件夹拿来安装，但是安装完之后，发现整个文件夹都消失了，我找了好久都没找到。最后，发现这个安装文件怎么跑到这个目录下面了：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1，如何有出现和我一样问题的，可以看看）

这里我提下，CUDA驱动并不是所有电脑都需要安装的，如果你的电脑里有如下这个图标，那么你的电脑很可能已经安装好 了CUDA。

2 下载cuDNN并且安装

在安装之前，看其他博客都说cuDNN的版本要和CUDA的版本相同，但是我的导师说他的CUDA版本与cuDNN版本不相同也可以，事实证明，确实可以！在官网下载cuDNN实在太慢了，所以我把我下载好的cuDNN-9.2版的上传上来，有需要的下载一下！（我的CUDA是10.1的，cuDNN是9.2的，实践出来这样有用）

链接：链接：https://pan.baidu.com/s/1Nzo2GLJCi9Dz0c9I90Ke