安装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即可。