終於!大家心心念念的PyTorch Windows官方支持來了
05-15
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 修復
推薦閱讀:
※花式解釋AutoEncoder與VAE
※Swift for TensorFlow 來啦!
※PyTorch框架快速增長,兩個月來GitHub新標星數居第三
※Win10 64位安裝Pytorch 0.3
※關於 PyTorch 0.3.0 在Windows下的安裝和使用