Windows下使用Anaconda 安装 Pytorch

现在 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

不报错就说明安装完成!!

这里写图片描述

  • 1
    点赞
  • 55
    收藏
    觉得还不错? 一键收藏
  • 12
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值