PyTorch 0.4.0 大更新，正式支持 Windows 平台

（點擊

上方公眾號

，可快速關注）

參考：PyTorch、開源中國，

整理：Python開發者（id：PythonCoder）

PyTorch 是一個 Python 優先的深度學習框架，提供兩個高級功能：

• 
  

  強大的 GPU 加速 Tensor 計算（類似 numpy）



• 
  

  構建基於 tape 的自動升級系統上的深度神經網路

4 月 25 日，PyTorch 的 GitHub 主頁宣布 PyTorch 0.4.0 已發布。

此次新版，除了平衡計算內存、支持更多概率分布、優化性能和修復 Bug 外，

PyTorch 還正式官方支持 Windows 系統

，不再需要藉助其它開發者發布的第三方 conda 包。

為 Python 3.5 和 3.6 提供預編譯的 Conda 二進位文件和 pip wheels。

另外，Windows 上的 PyTorch 不支持分散式訓練，可能比 Linux/OSX 慢一點，因為 Visual Studio 支持較早版本的 OpenMP。

Windows 用戶遇到問題，請看這個 FAQ：

http://pytorch.org/docs/stable/notes/windows.html

PyTorch

• 
  

  主要核心更改





• 
  

  Tensor / Variable 合併



• 
  

  Zero-dimensional Tensors



• 
  

  數據類型



• 
  

  遷移指南 （

  http://pytorch.org/2018/04/22/0_4_0-migration-guide.html

  ）

• 
  

  新特性





• 
  

  支持 RNN



• 
  

  易於使用的 Launcher utility



• 
  

  NCCL2 後端



• 
  

  24 個基礎的概率分布



• 
  

  新增 cdf, variance, entropy, perplexity 等



• 
  

  平衡計算內存



• 
  

  bottleneck - 一款識別代碼中的熱點的工具



• 
  

  全面支持高級索引



• 
  

  快速傅立葉轉換



• 
  

  Tensors



• 
  

  Neural Networks



• 
  

  torch.distributions



• 
  

  Distributed Training



• 
  

  C++ 拓展



• 
  

  支持 Windows



• 
  

  ONNX Improvements

具體細節，請查看 PyTorch 官方發行說明：

• 
  

  https://github.com/pytorch/pytorch/releases/tag/v0.4.0
  

看完本文有收穫？請轉

發分享給更多人

關注「P

ython開發者」，提升Python技能

喜歡這篇文章嗎？立刻分享出去讓更多人知道吧！