Auto Byte

专注未来出行及智能汽车科技

微信扫一扫获取更多资讯

Science AI

关注人工智能与其他前沿技术、基础学科的交叉研究与融合发展

微信扫一扫获取更多资讯

终于!大家心心念念的PyTorch Windows官方支持来了

五个小时前,PyTorch 官方 GitHub 发布 0.4.0 版本,大家心心念念的 Windows 支持终于来了。

这一次更新从 0.3.1 到 0.4.0 有着非常多的改进,最重要的就是官方支持 Windows 系统。在没有官方支持前,Windows 上安装 PyTorch 需要借助其它开发者发布的第三方 conda 包,而现在我们可以直接在 PyTorch 首页上获取使用 conda 或 pip 安装的命令行,或跟随教程使用源文件安装。

当然除了 Windows 支持外,0.4.0 版的 PyTorch 还重点权衡了计算中的内存、支持多种概率分布、优化数据类型和修正张量等。机器之心也尝试在 Windows 安装简单的 CPU 版,如下所示我们使用 pip 可以非常轻松而流畅地安装 PyTorch。

但当前使用 Conda 安装 PyTorch 会遇到一些问题,例如小编的 Conda 会报错说找不到对应的包。

此次版本更新目录如下:

1.重大核心变化包括:

  • Tensor/Variable 合并

  • 零维张量

  • 数据类型

  • 迁移指南

2.现添加的新特征包括:

  • Tensors:

    •     全面支持高级索引

    •     快速傅立叶变换

  • 神经网络:

    •     计算时的存储权衡

    •     bottleneck-识别代码中热点(hotspots)的工具

  • torch.distributions

    •     24 个基础的概率分布

    •     增加cdf、方差、信息熵、困惑度等

  • 分布式训练

    •     易于使用的 Launcher utility

    •     NCCL2 后端

  • C++拓展

  • Windows 支持

  • ONNX 改进

    •     RNN 支持

3.性能改进

4.Bug 修复

工程深度学习框架PyTorchWindows产品更新
1
暂无评论
暂无评论~