您的位置:首頁>正文

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

五個小時前, 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 修復

同類文章
Next Article
喜欢就按个赞吧!!!
点击关闭提示