windows python3.6环境下安装PyTorch0.4.0

文章参考：https://blog.csdn.net/xiangxianghehe/article/details/80103095

方法1：pip安装：

Python3.6+pip安装cpu版本

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

(需保证python是64位）

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 

方法2：Conda安装

如果你是Anaconda用户，执行命令：

conda install pytorch -c pytorch 

默认安装的是cuda8.0的gpu版本，如果你需要安装cuda9.0或者cuda9.1的版本，可以执行：

conda install pytorch cuda90 -c pytorch 

或者

conda install pytorch cuda91 -c pytorch 

测试是否安装成功：

import torch
print(torch.__version__)

如果输出0.4.0，则安装成功。

最后需要安装 torchvision：

pip install torchvision