pytorch 支持amd显卡吗_Windows系统安装和使用PyTorch 0.4.0 （官方已经支持 Windows ）...

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

发布时间：2018-04-25 20:52,

浏览次数：800

, 标签：

Windows

PyTorch

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 couldnot 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 arenot using fork to start your child processes and

you have forgotten to use the proper idiomin the main module: if __name__ ==

'__main__': freeze_support() ... The "freeze_support()" line can be omitted if

the programis not going to be frozen to produce an executable. #解决方案如下： import

torchdef 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