查看原文
其他

PyTorch 0.4.0 大更新,正式支持 Windows 平台

(点击上方公众号,可快速关注)


参考:PyTorch、开源中国,

整理:Python开发者(id:PythonCoder)


PyTorch 是一个 Python 优先的深度学习框架,提供两个高级功能:


  • 强大的 GPU 加速 Tensor 计算(类似 numpy)

  • 构建基于 tape 的自动升级系统上的深度神经网络

4 月 25 日,PyTorch 的 GitHub 主页宣布 PyTorch 0.4.0 已发布。


此次新版,除了平衡计算内存、支持更多概率分布、优化性能和修复 Bug 外,PyTorch 还正式官方支持 Windows 系统,不再需要借助其它开发者发布的第三方 conda 包。为 Python 3.5 和 3.6 提供预编译的 Conda 二进制文件和 pip wheels。


另外,Windows 上的 PyTorch 不支持分布式训练,可能比 Linux/OSX 慢一点,因为 Visual Studio 支持较早版本的 OpenMP。


Windows 用户遇到问题,请看这个 FAQ:

http://pytorch.org/docs/stable/notes/windows.html



PyTorch更新列表:


  • 主要核心更改

    • Tensor / Variable 合并

    • Zero-dimensional Tensors

    • 数据类型

    • 迁移指南 (http://pytorch.org/2018/04/22/0_4_0-migration-guide.html


  • 新特性

    • 支持 RNN

    • 易于使用的 Launcher utility

    • NCCL2 后端

    • 24 个基础的概率分布

    • 新增 cdf, variance, entropy, perplexity 等

    • 平衡计算内存

    • bottleneck - 一款识别代码中的热点的工具

    • 全面支持高级索引

    • 快速傅立叶转换

    • Tensors

    • Neural Networks

    • torch.distributions

    • Distributed Training

    • C++ 拓展

    • 支持 Windows

    • ONNX Improvements



具体细节,请查看 PyTorch 官方发行说明:


  • https://github.com/pytorch/pytorch/releases/tag/v0.4.0




看完本文有收获?请转发分享给更多人

关注「Python开发者」,提升Python技能


    您可能也对以下帖子感兴趣

    文章有问题?点此查看未经处理的缓存