Microsoft AI 开源“PyTorch-DirectML”:在 GPU 上训练机器学习模型的软件包

2021-11-05 11:18:14 浏览数 (1)

微软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/

0 人点赞