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

x
用微信扫描二维码
分享至好友和朋友圈

  机器之心整理

  参与:机器之心编辑部

  

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

  

  GitHub 发布https://github.com/pytorch/pytorch/releases

  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 修复

  

特别声明:本文为网易自媒体平台“网易号”作者上传并发布,仅代表该作者观点。网易仅提供信息发布平台。

跟贴 跟贴 1 参与 1
© 1997-2018 网易公司版权所有 About NetEase | 公司简介 | 联系方法 | 招聘信息 | 客户服务 | 隐私政策 | 广告服务 | 网站地图 | 意见反馈 | 不良信息举报

机器之心

专业的人工智能媒体

头像

机器之心

专业的人工智能媒体

2962

篇文章

79510

人关注

列表加载中...
请登录后再关注
x

用户登录

网易通行证/邮箱用户可以直接登录:
忘记密码