您当前的位置:首页 > 计算机 > 系统应用 > Windows

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

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

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()
我个人装的是Win10+GPU版本的,CUDA8+cuDNN v7,考虑到cuDNN需要注册才能下载,而且下载速度贼慢,提供两个百度云的下载链接,cuDNN安装教程见这个文章

Win10-x64+CUDA8+cuDNN v7

Win10-x64+CUDA8+cuDNN v6

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门