PyTorch安装

Windows

2018年4月25日，PyTorch 官方发布 0.4.0 版本，该版本的PyTorch 有多项重大更新，其中最重要的改进是支持 Windows 系统。Windows用户能直接通过conda、pip和源码编译三种方式来安装Pytorch，不过需要强调Windows下的Pytorch仅支持Python3.5和Python3.6，不支持其他的Python3版本，也不支持Python2。我们在官网就能看到相应的安装方式.

方法一：pip安装

Python3.6 + pip安装gpu版本
目前gpu版本支持cuda8.0,cuda9.0和cuda9.1，请选择对应的版本下载安装，不要同时执行下面三个命令！

pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl  
pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-win_amd64.whl 
pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl 

Python3.5+pip安装gpu版本
目前gpu版本支持cuda8.0,cuda9.0和cuda9.1，请选择对应的版本下载安装，不要同时执行下面三个命令！

pip install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl
pip install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-win_amd64.whl
pip install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp35-cp35m-win_amd64.whl

方法二：Conda安装

如果你是Anaconda用户，就不需要区分Python3.5和Python3.6，执行命令:

conda install pytorch -c pytorch 

就可以完成安装。不过这个默认安装的是cuda8.0的gpu版本,如果你需要安装cuda9.0或cuda1.0的gpu版本，请执行：

conda install pytorch cuda90 -c pytorch 

或者

conda install pytorch cuda91 -c pytorch 

来进行安装。

PS: 我在用conda(Python 3.5中进行安装时)，无法正常下载，采用pip进行安装，可以正常安装.

测试安装是否成功

import torch
print(torch.__version__)

如果输出0.4.0，那么恭喜Windows下的PyTorch0.4.0安装成功！

最后需要安装 torchvision：

pip install torchvision

一般gpu版本配置需要cuDNN，而cuDNN需要注册才能下载，下载速度太慢，有网友提供两个百度云的下载链接，cuDNN安装教程见这里。

Win10-x64+CUDA8+cuDNN v7
Win10-x64+CUDA8+cuDNN v6

更新pip到pip3

pytorch 1.0安装时用conda无法完成下载，只能用pip安装

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

，但是pytorch 1.0版本需要pip3版本，需要将pip升级到pip3，有两种方式:

1. python -m pip install --upgrade pip
2. conda install mingw libpython
   网上大多数的文章都推荐第二种方法，但是我无法使用，用第一种方法完成升级(记得要在Anaconda3的Prompt中执行)