win查看cuda版本_windows下pytorch1.0(gpu版本）安装

pytorch官方入门视频教程

https://in.udacity.com/course/deep-learning-pytorch--ud188​in.udacity.com

1. pip安装

• 查看自己的CUDA版本

如果没有CUDA请安装CUDA和CUDNN（安装可参照此链接JustDoIT：windows tensorflow-gpu的安装），如果有CUDA且知道自己的版本请跳过这一步骤

按照上图的顺序即可查看CUDA的版本

• 从pytorch(PyTorch)官网下载对应的whl文件并安装

按照上图的顺序即可，可以复制那个https的链接到浏览器，先下载whl文件，然后再用pip命令安装（推荐这种方法）,如果网速好的话可以直接复制上面的那两条pip命令即可

如果是先下载whl文件再安装，就使用下面的命令（我的whl文件路径是D:torch-1.0.0-cp35-cp35m-win_amd64.whl）

pip3 install D:torch-1.0.0-cp35-cp35m-win_amd64.whl
pip3 install torchvision

如果是直接安装就使用下面的命令

pip3 install https://download.pytorch.org/whl/cu90/torch-1.0.0-cp35-cp35m-win_amd64.whl
pip3 install torchvision

• 测试pytorch GPU版本是否安装成功

import torch
print(torch.cuda.is_available())
#输出true则安装成功

除了官网，也可以在下面这个网站学习PyTorch

PyTorch 中文网 - PyTorch 中文教程网​www.pytorchtutorial.com