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