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,有两种方式:
- python -m pip install --upgrade pip
- conda install mingw libpython
网上大多数的文章都推荐第二种方法,但是我无法使用,用第一种方法完成升级(记得要在Anaconda3的Prompt中执行)