Windows系统安装和使用PyTorch 0.4.0 （官方已经支持 Windows ）

时间：03-27来源：作者：点击数：24

PyTorch 0.4.0 官方已经支持 Windows 系统

2018年4月25号，PyTorch 官方发布 0.4.0 版本，该版本的 PyTorch 有多项重大更新，其中最重要的改进是官方支持 Windows (详细改动见Pytorch官方GitHub)：

Tensor/Variable 合并
零维张量
数据类型
迁移指南
Windows 支持
C++ 扩展
ONNX 改进支持 RNN
Bug修复与性能优化

Pytorch官网新增安装方式——Windows安装

Windows各种安装方式汇总：

#使用conda安装，不支持python2.7,cudanone就是CPU版本的
'conda,cuda8,python3.5': conda install pytorch -c pytorch
'conda,cuda9.0,python3.5': conda install pytorch cuda90 -c pytorch
'conda,cuda9.1,python3.5': conda install pytorch cuda91
'conda,cudanone,python3.5': conda install pytorch-cpu -c pytorch
'conda,cuda8,python3.6': conda install pytorch -c pytorch
'conda,cuda9.0,python3.6': conda install pytorch cuda90 -c pytorch
'conda,cuda9.1,python3.6': conda install pytorch cuda91 -c pytorch
'conda,cudanone,python3.6': conda install pytorch-cpu -c pytorch

#使用pip安装，不支持python2.7,cudanone就是CPU版本的
'pip,cudanone,python3.5': pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp35-cp35m-win_amd64.whl
'pip,cuda8,python3.5': pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp35-cp35m-win_amd64.whl
'pip,cuda9.0,python3.5': pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp35-cp35m-win_amd64.whl
'pip,cuda9.1,python3.5': pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp35-cp35m-win_amd64.whl
'pip,cudanone,python3.6': pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.0-cp36-cp36m-win_amd64.whl
'pip,cuda8,python3.6': pip3 install http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-win_amd64.whl
'pip,cuda9.0,python3.6': pip3 install http://download.pytorch.org/whl/cu90/torch-0.4.0-cp36-cp36m-win_amd64.whl
'pip,cuda9.1,python3.6': pip3 install http://download.pytorch.org/whl/cu91/torch-0.4.0-cp36-cp36m-win_amd64.whl

#源码编译
需要安装VS2017，我电脑没有，所以略过。

Windows用户能直接通过conda、pip和源码编译三种方式来安装Pytorch了，假如我们用pip来安装的话：

安装GPU版本，如：Windows+pip+python 3.6+CUDA 8 ，只需在cmd中输入：

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

如果电脑没有英伟达显卡，那就只能安装CPU版本的，如：Windows+pip+python 3.6 ，在cmd中输入：

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

最后需要安装 torchvision，我是通过本地安装的方式进行安装的，下载地址：（https://pypi.python.org/pypi/torchvision/0.2.1）下载好之后，用cmd进入torchvision所在的文件夹下，输入下面的命令安装：

pip install torchvision-0.2.1-py2.py3-none-any.whl

测试一下是否安装好了，看看能不能用GPU加速：

>>>import torch
>>>torch.__version__
>>>torch.cuda.is_available()

有几个小伙伴反馈安装后会出现一些问题，官方文档针对一些问题给出了解决方案，在知乎上有大神也给出了一些解决方案。

1.import torch失败

#import torch时出现以下错误：
from torch._C import *

ImportError: DLL load failed: The specified module could not be found.

#解决方案：缺少了必要的一些动态链接库，或者没英伟达显卡而误装了GPU版本的
在cmd中输入：conda install -c peterjc123 vc vs2017_runtime

2.运行错误

#使用pytorch时出现的多进程错误，如下描述：
RuntimeError:
    An attempt has been made to start a new process before the
    current process has finished its bootstrapping phase.

   This probably means that you are not using fork to start your
   child processes and you have forgotten to use the proper idiom
   in the main module:

       if __name__ == '__main__':
           freeze_support()
           ...

   The "freeze_support()" line can be omitted if the program
   is not going to be frozen to produce an executable.

#解决方案如下：
import torch
def main()
    for i, data in enumerate(dataloader):
        # do something here
if __name__ == '__main__':
    main()