五個小時前, 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 支持外, 0.4.0 版的 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 修復