【安装教程】win10+GPU+pytorch安装教程

Windows10下GPU版本的Pytorch安装教程

1.查看自己的显卡版本型号：
打开显卡控制面板–左下角的系统信息–组件–查看自己的显卡支持的cuda版本型号
2.下载cuda
我选择的是9.2版本的cuda，因此9.2版本的cuda官网下载：https://developer.nvidia.com/cuda-92-download-archive
根据自己的显卡支持的版本，选择对应的版本型号如下，并点击download开始下载

3.下载cudnn：
1.首先需要注册一个账号

2.官网下载：https://developer.nvidia.com/rdp/cudnn-download
3.选择对应的版本及系统：

4.安装cuda
按照提示进行即可可以选择默认路径或者自定义
5.测试
测试cuda是否安装成功，找到cuda安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\bin，cmd打开prompt，cd到指定路径下，执行nvcc -V指令，出现如下结果证明安装成功

6.cudnn处理
将文件夹解压，并将里面的三个文件夹复制黏贴到cuda安装目录下

7测试
进入demo文件中进行两个程序的测试C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.2\extras\demo_suite，执行bandwidthTest.exe及deviceQuery.exe，显示pass则安装成功

8.pytorch安装
采用conda安装，但是比较慢，预计半个多小时

将conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev复制 直接conda安装
9.pytorch检测
打开pycharm 输入import Torch 无报错则安装成功。