微软开源可在任何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训练脚本。