Windows 版 PyTorch 開發(fā)維護權(quán)已移交至微軟
Facebook 通過 PyTorch 官方博客宣布,微軟已擴大了對 PyTorch 社區(qū)的參與,現(xiàn)在擁有 PyTorch 在 Windows 上的開發(fā)和維護所有權(quán)。
PyTorch 是基于 Torch 的開源 Python 機器學習庫,由 Facebook 的人工智能研究團隊開發(fā)。特斯拉的自動駕駛和 Uber 的 Pyro 等深度學習項目都構(gòu)建于 PyTorch 之上。
此前,F(xiàn)acebook 的工程師們做了大量努力,在 Windows 上添加了對 PyTorch 的支持。但由于資源有限,Windows 對 PyTorch 的支持落后于其他平臺。例如,Windows 上缺少對 PyTorch 的官方 PyPI 支持,用戶的安裝體驗不太流暢;某些 PyTorch 功能在 Windows 平臺上也根本不可用,像是 TorchAudio 域庫和分布式訓練支持等等。微軟的加入將會在很大程度上改進這一局面。
隨著 PyTorch 1.6 版本的發(fā)布,通過使測試覆蓋范圍與 Linux 的 PyTorch 核心及其域庫保持一致,并使教程測試自動化,Windows 版本的核心質(zhì)量大幅提高。官方表示在之后的版本中將繼續(xù)改進 Windows 上 PyTorch 的體驗,接下來的重點,也是社區(qū)中呼聲最高的,是對分布式訓練的支持,以及更好的 pip 安裝體驗。
在今年的 Build 2020 大會上,微軟宣布計劃為 WSL 提供 GPU 計算支持,PyTorch 將在其中起到不小的作用?,F(xiàn)階段,WSL2 已獲得對 GPU 的初始支持,其中就包含對 PyTorch 的支持,WSL 用戶可以直接運行本地 PyTorch 程序,進行機器學習。
微軟擴大對 PyTorch 的支持,成為其開發(fā)和維護者,無疑也將為 WSL 帶來更好的用戶體驗。