終於!大家心心念念的PyTorch Windows官方支持來了

PyTorch 官方 GitHub 發布 0.4.0 版本,大家心心念念的 Windows 支持終於來了。

  • GitHub 發布地址:github.com/pytorch/pyto
  • PyTorch 官網: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 修復


推薦閱讀:

花式解釋AutoEncoder與VAE
Swift for TensorFlow 來啦!
PyTorch框架快速增長,兩個月來GitHub新標星數居第三
Win10 64位安裝Pytorch 0.3
關於 PyTorch 0.3.0 在Windows下的安裝和使用

TAG:PyTorch | 人工智慧 |