现在 Pytorch 0.4 版本以后已经开始支持Windows系统了,下面是基于官网给出的教程,方便不能翻墙的同学使用:
Tips: 这里的 Anaconda一个是 anaconda3,即python 的版本是 3版本的,这里 windows 下 pytorch 不在支持 python2.7。
不论是 pip 的安装方式
还是 conda 的安装方式:
这里以 python3.6 版本为例,若为3.5,或者3.7版本的,conda 安装方式没有变化,只是将pip 方式的安装 xxx-cp36-cp36m-xxx,手动换成对应的xxx-cp35-cp35m-xxx或者xxx-cp37-cp37m-xxx就可以了。
————————————————————————— 分割线————————————————————————
(2019.4.21 日更新)
写博文的时候最新版本还是 0.4,现在 1.0 已经出来了,修改版本信息同上,如果想安装 1.0 版本的,可以手动将 pip 安装方式的 torch-0.4.1-xxx 替换成 torch-1.0.1-xxx 即可。
—————————————————————————————————————————————————————
请根据自己电脑中的相关配置信息选择下列安装方式(conda或者pip 任选其一即可):
(windows 下 CUDA + cudnn 安装方法见:Windows 下用 build_win.cmd 直接编译 GPU 版caffe 1、2 部分)
安装方法:
1. 没有安装CUDA(或者没有GPU)
conda 方式
conda install pytorch-cpu -c pytorch
pip install torchvision
pip 方式
pip install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip install torchvision
2. CUDA8.0
conda 方式
conda install pytorch cuda80 -c pytorch
pip install torchvision
pip 方式
pip install http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip install torchvision
3. CUDA9.0
conda 方式
conda install pytorch cuda90 -c pytorch
pip install torchvision
pip 方式
pip install http://download.pytorch.org/whl/cu90/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip install torchvision
4. CUDA9.2
conda 方式
conda install pytorch cuda92 -c pytorch
pip install torchvision
pip 方式
pip install http://download.pytorch.org/whl/cu92/torch-0.4.1-cp36-cp36m-win_amd64.whl
pip install torchvision
可能遇到下载失败的情况
以 Pytorch 0.4.1 为例,安装包大小是 535.5 MB,在下载过程中只能看到进度条,可能下载过程中因为网络原因,出现下载失败的情况,这种通常情况下只能重新运行命令,继续从头下载,也可能多次失败。
1. conda 方式
2. pip 方式
(这里下载速度只有 几十 KB/s 的速度,我就手动停止了),也可以看出 pip 安装方式是可以实时看到当前下载速度和进度条的,而 conda 安装方式只可以看到进度条。
解决方法
一般我遇到上述情况,都是选择将 pip install xxx,这里的 xxx 代表上面的下载链接,直接将链接复制到浏览器中,使用浏览器来下载,以 CUDAO8.0+3.5 版本的为例,3.6或3.7 的只要将对用 xxx–cp35-cp35m-xxx 换成 xxx–cp36-cp36m-xxx,xxx–cp37-cp37m-xxx 就可以了。
打开浏览器输入:
http://download.pytorch.org/whl/cu80/torch-0.4.1-cp36-cp36m-win_amd64.whl
然后回车:
若在下载中途出现问题,还可以点 继续 下载,而不是每次都从头下载,并且安装包可以多次使用。
下载完成后,在对应的 ,whl 下载路径下打开命令行工具输入:
pip install torch-0.4.1-cp35-cp35m-win_amd64.whl
稍等就可以完成安装了。
安装完可以打开 Python 环境输入:
import torch
不报错就说明安装完成!!