pytorch零基础教程(PyTorch官方终于支持Windows了)

pytorch零基础教程(PyTorch官方终于支持Windows了)(1)

PyTorch今天发布了v0.4.0版本,网友们反响非常热烈,甚至有人说:感觉就像提前过圣诞~

这次的更新中,最受关注的有两点。一是Tensors(张量)和Variables(变量)的合并;二是官方终于支持Windows系统了。

pytorch零基础教程(PyTorch官方终于支持Windows了)(2)

以下是本次更新的主要内容:

重大核心变化:

  • Tensors和Variables合并了

  • 有的运算会返回零维张量(标量)

  • volatile flag作废了

  • 可以通过torch.dtype、torch.device和torch.layout更好地管理数据类型属性

  • 支持编写不区分设备的代码

除此之外,PyTorch在张量、神经网络、分布式训练等方面增加了一些新功能,包括:

  • 张量

    • 全面支持高级索引

    • 支持快速傅立叶变换

  • 神经网络

    • 新增了一个checkpoint容器,可以在内存使用和计算速度之间折衷

    • 新增了一个名叫瓶颈的工具,可以对程序中的瓶颈进行初步debug

  • torch.distributions

    • 扩展到包含24个基本概率分布

    • 增加了cdf,方差,熵,复杂度等计算

  • 分布式训练

    • 增加了易用的启动器

    • 提供基于NCCL2的新分布式后端

  • 为自己编写C 或CUDA扩展提供了更好的系统

  • 官方支持Windows,不过Windows版不支持分布式训练,可能比Linux版慢一点

  • 改进了ONNX的支持

  • 提供更好的RNN支持

在这次更新前不久,PyTorch还发生过一次重大变化:Caffe 2源代码全部并入了PyTorch,Facebook这大深度学习框架合二为一。

pytorch零基础教程(PyTorch官方终于支持Windows了)(3)

最后,要用在v0.3.x之前写的旧版代码,一定要看一眼官方迁移指南哦~

http://pytorch.org/2018/04/22/0_4_0-migration-guide.html

— 完 —

诚挚招聘

量子位正在招募编辑/记者,工作地点在北京中关村。期待有才气、有热情的同学加入我们!相关细节,请在量子位公众号(QbitAI)对话界面,回复“招聘”两个字。

量子位 QbitAI · 头条号签约作者

վ'ᴗ' ի 追踪AI技术和产品新动态

,

免责声明:本文仅代表文章作者的个人观点,与本站无关。其原创性、真实性以及文中陈述文字和内容未经本站证实,对本文以及其中全部或者部分内容文字的真实性、完整性和原创性本站不作任何保证或承诺,请读者仅作参考,并自行核实相关内容。文章投诉邮箱:anhduc.ph@yahoo.com

    分享
    投诉
    首页