微软开源可在任何GPU训练机器学习模型的组件PyTorch-DirectML

微软的Windows人工智能团队发布可作为PyTorch后端,用来训练机器学习模型的DirectML预览版,官方提到,这是他们要在任何Windows DirectX12 GPU上以及WSL(Windows Subsystem for Linux),支持PyTorch机器学习训练的一步进展。

借由这个新发布的PyTorch-DirectML组件,开发者可以在PyTorch中利用DirectML,以及相关卷积神经网络支持。微软提到,PyTorch-DirectML通过在DirectML中加入名为DML的新设备,来跟PyTorch框架集成,该设备能够调用DirectML API和PyTorch Tensor原语。

这项成果是由微软、AMD、英特尔和Nvidia共同合作设计,对PyTorch提供硬件加速的功能,官方提到,他们最小化调用DirectML运算符的资源开销,并且使得DirectML后端的运行方式,和现有PyTorch后端相似,而且PyTorch-DirectML组件的安装方法也相当简单,只需要更改现有脚本的一行程序代码,就可以开始执行PyTorch训练脚本。