微软Windows团队的AI已经公布了˚F IRST DirectML的预览作为后端PyTorch训练ML车型。此版本允许在任何 DirectX12 GPU 和 WSL 上加速 PyTorch 的机器学习训练,释放混合现实计算的新潜力。
微软 AI 团队与 PyTorch 框架合作发布了一个预览包,为 CNN(卷积神经网络)提供范围支持。在这个名为“DML”的新设备中,通过在调用运算符时引入最少的开销来调用直接 ML API 和 Tensor 原语;它们的工作方式与其他现有后端非常相似。
PyTorch-DirectML 套件可以使用 GPU 机器学习库 DirectML 作为其后端,允许在 DirectX12 GPU 和 WSL(适用于 Linux 的 Windows 子系统)上训练模型。
Microsoft 与 AMD、Intel 和 NVIDIA 合作,为 PyTorch 提供这种硬件加速的训练体验。PyTorch-DirectML 包安装简单,只需更改现有脚本中的一行代码。
Github:
https://github.com/microsoft/DirectML/
参考:
https://devblogs.microsoft.com/windowsai/introducing-pytorch-directml-train-your-machine-learning-models-on-any-gpu/