Windows 现在通过 DirectML 原生支持 PyTorch

与为图形提供低级 API 的 DirectX 类似，DirectML 为 Windows 中的机器学习提供低级 API。 DirectML 允许开发人员一次构建并使其可用于所有支持的 Windows PC，而不是浪费时间和精力来针对不同的 CPU、GPU 和 NPU 硬件选项。它还与 ONNX Runtime、PyTorch 和 WebNN 等框架集成。

在 Build 2024 大会上，微软今天 公布 Windows 现在通过 DirectML 原生支持 PyTorch。借助这种原生 PyTorch 支持，数千个 Hugging Face 模型无需任何更改即可在 Windows 上运行。此外，微软正在与 Nvidia 合作，使这些开发工作流程在超过 100M RTX AI GPU 中得到支持。 PyTorch 对 GPU 的支持从今天开始提供，NPU 支持也即将推出。

微软还通过 WebNN Developer Preview 为 Web 应用程序启用本地 AI 的强大功能。 WebNN 由 DirectML 和 ONNX Runtime Web 提供支持。 WebNN 允许开发人员为生成式 AI、图像处理、自然语言处理、计算机视觉等任务提供接近本机性能的 AI 体验。目前，WebNN 开发者预览版支持
GPU。 NPU 支持即将推出。

DirectML 可用性：

• DirectML 现在普遍适用于所有 Windows GPU。
• 具有 Intel AI Boost 的 Intel Core Ultra 处理器上的 DirectML 支持作为开发者预览版提供，即将全面上市
• Snapdragon X Elite SoC 中的 Qualcomm Hexagon NPU 即将推出 DirectML 支持。