输入关键词搜索 AI 工具、分类,或直接跳转页面
DirectML是微软推出的面向Windows平台的机器学习底层API,核心定位是为各类硬件加速器提供统一的抽象适配层,降低AI功能在Windows生态内的集成门槛。核心功能包含多硬件统一适配能力,支持DirectX 12兼容的GPU、NPU等异构计算设备接入,开发者无需针对不同硬件编写差异化驱动适配代码;内置丰富的算子库,覆盖图像识别、自然语言处理、大模型推理等主流AI任务的算子需求;支持与现有主流机器学习框架无缝对接,可直接适配PyTorch、TensorFlow等工具的工作流。目标用户覆盖Windows平台AI应用开发者、硬件驱动开发人员、游戏AI功能开发工程师、边缘AI部署工程师等。适用于桌面端AI应用开发、游戏内AI功能嵌入、端侧大模型推理优化、硬件厂商加速器适配开发等多种场景。
访问微软DirectX开发者博客的DirectML介绍页面,下载对应版本的DirectML SDK安装包,按照页面提示完成开发环境的基础配置,确认本地开发设备支持DirectX 12及以上版本。
查看官方提供的快速上手示例代码,根据自身开发需求选择对应的算子调用、模型部署示例项目,将示例代码导入到本地的Visual Studio开发环境中。
若需要对接现有机器学习框架,可在框架的后端设置选项中选择DirectML作为推理后端,导入已训练完成的模型文件,测试基础推理功能是否可正常运行。
根据业务需求开发自定义的AI功能逻辑,若需要适配NPU硬件,可在代码中指定NPU作为优先计算设备,测试功能在不同硬件环境下的运行效果。
完成功能开发后,使用官方提供的性能分析工具检测推理效率和资源占用情况,针对瓶颈环节进行优化,最终将集成DirectML的应用打包发布到Windows平台。
看完教程,直接上手试试
访问 DirectML 官网