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

Posted 2021-05-02 机器之心

机器之心整理

参与：机器之心编辑部

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

• PyTorch 官网：http://pytorch.org/

这一次更新从 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 修复

蚂蚁金服举办首届金融科技领域算法类大赛——ATEC 蚂蚁开发者大赛人工智能大赛，点击「阅读原文」 进入大赛官网了解比赛信息，比赛报名请使用PC端浏览器打开官网。