2017年1月18日,周董生日这一天,facebook下的torch7团队宣布Pytorch开源,官网地址:pytorch。pytorch是一个python优先的深度学习框架,是一个和tensorflow,Caffe,MXnet一样,非常底层的框架。先说下pytorch的三大优势:
一.Python优先支持策略
Pytorch主推的特性之一,就是支持python(官方的提法:puts Python first)。因为直接构建自 Python C API,Pytorch从细粒度上直接支持python的访问。相比于原生Python实现,引入的新概念很少,这不仅降低了 python 用户理解的门槛,也能保证代码基本跟原生的 python 实现一致。事实上,开发者可以直接用原生 python 代码扩展 Pytorch 的 operation。
二.动态图的良好支持
Tensorflow运行必须提前建好静态计算图,然后通过feed和run重复执行建好的图。但是Pytorch却不需要这么麻烦:PyTorch的程序可以在执行时动态构建/调整计算图。相对来说,pytorch具有更好的灵活性。这得益于Pytorch直接基于 python C API 构建的 python 接口。
三.易于Debug
Pytorch在运行时可以生成动态图,开发者就可以在堆栈跟踪中看到哪一行代码导致了错误。你甚至可以在调试器中停掉解释器并看看某个层会产生什么。
好了,吹完了Pytorch这么多的好处,想安装体验一把,结果上官网一看,Pytorch不支持windows!!!
这就没得玩了?不!
好了,开始正题,如何在win10 python3.6开启Pytorch支持:首先安装Anaconda3,然后下载Pytorch的conda 安装包,地址:pytorch conda安装包—pytorch-0.1.12-py36_0.1.12cu80.tar.bz2 下载完毕之后,直接cmd进入到pytorch-0.1.12-py36_0.1.12cu80.tar.bz2对应的目录,执行以下命令:
conda install numpy mkl cffi
conda install --offline pytorch-0.1.12-py36_0.1.12cu80.tar.bz2 #离线安装
然后python命令下输入:
import torch
如果不报错,就是安装成功!
本教程在win10和Python 3.6.0 |Anaconda 4.3.1 (64-bit)环境下安装成功,如果安装失败,请在本文评论或私信本作者。
已知安装成功环境 :
1. Windows (7/8/8.1/10)64位 + Anaconda-4.3.1 (64-bit) Python 3.6.0
2. Windows (7/8/8.1/10)64位 + Anaconda-4.4.0 (64-bit) Python 3.6.1
3. Windows (7/8/8.1/10)64位 + Anaconda-5.0.1 (64-bit) Python 3.6.3
Anaconda中国镜像下载地址 :https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive。
原文地址:PyTorch在64位Windows下的Conda包:https://zhuanlan.zhihu.com/p/26871672。
本文系转载,对于未能及时标注转载出处深感歉意。
2017.11.21更新:统一回复:Windows 64位环境下未安装成功的同学请卸载旧的Anaconda3版本,安装最新版的Anaconda3-5.0.1的64位版本,然后进入到pytorch-0.1.12-py36_0.1.12cu80.tar.bz2文件的目录,在控制台下直接执行:
conda install --offline pytorch-0.1.12-py36_0.1.12cu80.tar.bz2
即可安装成功!
点击Anaconda3-5.0.1下载地址,拉到最下面,就能找到Anaconda3-5.0.1-Windows-x86_64.exe。
**2017.12.07更新:**本文已过期,Windows安装最新的PyTorch0.3.0请移步本人另一篇博客:Windows下安装PyTorch0.3.0
**2018.4.27更新:**本文已过期,PyTorch0.4.0已经有官方的Windows支持,Windows下安装最新的PyTorch0.4.0请移步本人另一篇博客:Windows下安装PyTorch0.4.0