一、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/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,顺序安装就好
四、Pytorch
我选择的是如下版本,
在Anaconda命令中输入“conda install pytorch cuda91-c pytorch”便可开始下载
检验:
至此,完成
更多推荐
win10下安装Pytorch (CUDA9.1+CUDNN7.0)
发布评论