Windows 版 PyTorch 開發(fā)維護(hù)權(quán)已移交至微軟

Facebook 通過 PyTorch 官方博客宣布，微軟已擴(kuò)大了對(duì) PyTorch 社區(qū)的參與，現(xiàn)在擁有 PyTorch 在 Windows 上的開發(fā)和維護(hù)所有權(quán)。

PyTorch 是基于 Torch 的開源 Python 機(jī)器學(xué)習(xí)庫(kù)，由 Facebook 的人工智能研究團(tuán)隊(duì)開發(fā)。特斯拉的自動(dòng)駕駛和 Uber 的 Pyro 等深度學(xué)習(xí)項(xiàng)目都構(gòu)建于 PyTorch 之上。

此前，F(xiàn)acebook 的工程師們做了大量努力，在 Windows 上添加了對(duì) PyTorch 的支持。但由于資源有限，Windows 對(duì) PyTorch 的支持落后于其他平臺(tái)。例如，Windows 上缺少對(duì) PyTorch 的官方 PyPI 支持，用戶的安裝體驗(yàn)不太流暢；某些 PyTorch 功能在 Windows 平臺(tái)上也根本不可用，像是 TorchAudio 域庫(kù)和分布式訓(xùn)練支持等等。微軟的加入將會(huì)在很大程度上改進(jìn)這一局面。

隨著 PyTorch 1.6 版本的發(fā)布，通過使測(cè)試覆蓋范圍與 Linux 的 PyTorch 核心及其域庫(kù)保持一致，并使教程測(cè)試自動(dòng)化，Windows 版本的核心質(zhì)量大幅提高。官方表示在之后的版本中將繼續(xù)改進(jìn) Windows 上 PyTorch 的體驗(yàn)，接下來的重點(diǎn)，也是社區(qū)中呼聲最高的，是對(duì)分布式訓(xùn)練的支持，以及更好的 pip 安裝體驗(yàn)。

在今年的 Build 2020 大會(huì)上，微軟宣布計(jì)劃為 WSL 提供 GPU 計(jì)算支持，PyTorch 將在其中起到不小的作用?，F(xiàn)階段，WSL2 已獲得對(duì) GPU 的初始支持，其中就包含對(duì) PyTorch 的支持，WSL 用戶可以直接運(yùn)行本地 PyTorch 程序，進(jìn)行機(jī)器學(xué)習(xí)。

微軟擴(kuò)大對(duì) PyTorch 的支持，成為其開發(fā)和維護(hù)者，無疑也將為 WSL 帶來更好的用戶體驗(yàn)。