win10下安装Pytorch (CUDA9.1+CUDNN7.0)

一、CUDA

下载地址，直接搜索便可找到

我下载的是9.1版本的

安装步骤：

（1）点击cuda_9.1.85_win10.exe安装

（2）按顺序同意就好

（3）完成 得到如下两个图标

（4）官网下载时还有以下三个patch，反正我又顺次点开安装了一遍，没仔细看是做什么的

（5）变量配置

（6）检验。

①win+R，打开cmd窗口，输入nvcc –V，显示CUDA版本信息，安装成功。

②用Visual studio编译C:\ProgramData\NVIDIACorporation\CUDA Samples \v9.1\ 1_Utilities\ deviceQuery下的deviceQuery_vs2015.sln文件，生成C:\ProgramData\NVIDIACorporation\CUDA Samples\v9.1\bin\win64\Release下的deviceQuery.exe，在cmd中运行，显示PASS则成功。

二、CUDNN

下载链接：https://developer.nvidia.com/rdp/cudnn-archive

打开压缩包，依次将压缩包中的3个文件复制到相应位置

1.    C:\cuda\bin\cudnn64_7.dll —> C:\Program Files\NVIDIA GPUComputing 
Toolkit\CUDA\v9.1\bin

2.    C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPUComputing 
Toolkit\CUDA\v9.1\include

3.    C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPUComputing 
Toolkit\CUDA\v9.1\lib\x64

三、Visual Studio 2017

官网可直接下载https://www.microsoft.com，顺序安装就好

四、Pytorch

我选择的是如下版本，

在Anaconda命令中输入“conda install pytorch cuda91-c pytorch”便可开始下载 

检验：

至此，完成