直接机器学习 (DirectML)

直接机器学习 (DirectML) 是机器学习的低级 API。 它具有常见的(本机 C++、nano-COM)编程接口和 DirectX 12 样式的工作流。 可将机器学习推断工作负荷集成到游戏、引擎、中间件、后端或其他应用程序中。 所有与 DirectX 12 兼容的硬件都支持 DirectML。

DirectML 是在 Windows 10 版本 1903 和相应版本的 Windows SDK 中引入的。

请务必查看登陆页面

本节内容

主题 说明
DirectML 简介 直接机器学习 (DirectML) 是机器学习 (ML) 的低级 API。
DirectML 版本历史记录 DirectML 是 Windows 10 的系统组件,也可作为独立的可再发行组件包提供。
DirectML 功能级别历史记录 每个功能级别中引入的类型清单。
DirectML 中的绑定 在 DirectML 中,绑定是指将资源附加到管道,以供 GPU 在机器学习运算符初始化和执行时使用。 例如,这些资源可以是输入和输出张量,也可以是运算符需要的任何临时性或永久性资源。
DirectML 中的 UAV 屏障和资源状态屏障 描述屏障的正确性好处,以及在 DirectML 中的使用方式。
资源生存期和同步 DirectML 应用程序必须准确地管理对象生存期和 CPU 与 GPU 之间的同步,以避免未定义的行为。
使用步幅来表示填充和内存布局 DirectML 张量用张量的“大小”和“步幅”等属性进行描述
使用融合运算符提高性能 某些 DirectML 运算符支持称为“融合”的概念。 运算符融合是提高性能的一种方法,通过将一个运算符(通常是激活函数)合并到另一个运算符以便其一起执行,而无需往返内存。
使用 DirectML 调试层 DirectML 调试层是可选的开发时组件,可帮助调试 DirectML 代码。
处理错误和设备删除 本主题论述如何调试 DirectML 设备删除和其他错误条件。
DirectMLX DirectMLX 是 DirectML 的 C++ 仅标头帮助程序库,旨在更轻松地将单个运算符组合到图形中。
DirectML 帮助程序函数 基础 DirectML 帮助程序函数的代码列表。
DirectML 示例应用程序 链接到 DirectML 示例应用程序,其中包括最小的 DirectML 应用程序示例。
GPU 加速 ML 训练 介绍适用于 Linux 的 Windows 子系统 (WSL) 和本机 Windows 的 GPU 加速机器学习 (ML) 训练当前支持的内容。
DirectML API 参考 本节介绍在 DirectML.h 中声明的 Direct 机器学习 (DirectML) API。