win10安装gpu版pytorch

安装CUDA和cudnn

官网下载太慢，建议其他地方下载，CUDA是NVIDIA的运算平台，cudnn是用于深度神经网络的gpu加速库。CUDA10一般的N卡都可以，我用的CUDA9，建议下载CUDA10，安装目录我直接默认c盘，两个多G，安装后cmd里 nvcc -V测试成功否。
我参照的是这里
据说安装之前必须要vs，我原来就安的vs2017。

安装pytorch

之后安装pytorch，用清华镜像比较快，十分钟就好了
第一步添加源 conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
第二步 conda install pytorch1.0.1 torchvision0.2.2
这里两个的版本都可更改，可自己网上查对应的版本。
最后安完之后测试

import torch
import torchvision
print(torch.cuda.is_available())

结果True则安装成功。

记录一下我安装的时候遇到的问题
CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://mirrors.ustc.edu.cn/anaconda/pkg

conda安装时一直报错，换源什么的都不好使，折腾了半天，直到看到https://blog.csdn.net/u013383596/article/details/87718472

将https改为http即可。