微软将接手PyTorch for Windows维护

PyTorch项目本周宣布微软将接手Windows版本PyTorch框架的维护工作。

PyTorch是脸书于2017年开源的机器学习框架及脚本程序语言。在市场上的众多框架中,PyTorch受开发人员欢迎程度仅次于Google TensorFlow,也是去年增长最快的开源项目。

根据统计,Windows仍是最主要开发平台(46% Windows vs 28% MacOS),然而使用Windows版PyTorch的开发人员仍落后于其他平台,原因包括测试覆盖率较少导致问题层出不穷、核心教学文件不齐全、PyPI不支持PyTorch for Windows造成安装不易等。此外有些PyTorch功能,像是TorchAudio函数库及分布式训练支持等也不支持Windows。微软将接手维护PyTorch for Windows以解决这些问题。

微软此举旨在让Windows 10 PC及WSL环境下提升PyTorch执行性能,最终目的是让开发人员可在WSL环境下执行机器学习训练。

另一方面,6月间,微软公布WSL 2加入GPU支持,也是为了开发人员可在Windows 10硬件上执行AI和机器学习应用,并获得了AMD、英特尔和Nvidia的支持。其中微软和Nvidia联合推出的Nvidia CUDA for WSL 2预览版,即支持PyTorch及TensorFlow。

PyTorch项目今天也发布PyTorch 1.6,新版本将扩大Windows版核心PyTorch及其函数库的测试覆盖率和Linux版相当,其中后者将波及TorchVision、TorchText及TorchAudio。

PyTorch项目人员表示,依据开发社群的意见,分布式训练支持及以pip安装,为下一阶段的改进目标。