去年 6 月，微软宣布它正在与硬件供应商合作一个名为 TensorFlow-DirectML 的项目，该项目可以在 Windows Subsystem for Linux（WSL）上提供 GPU 加速的机器学习（ML）模型的训练。经过了一年时间的预览，该项目如今已结束了预览，发布了正式版本。
![]()
顾名思义，TensorFlow-DirectML 是 TensorFlow 的一个分叉，它既包括了 Google 的 TensorFlow 机器学习平台，也包括了微软的 DirectML API。开发者能够利用 TensorFlow-DirectML 为在 Windows 10 和 WSL 上训练 ML 模型获得跨厂商的硬件加速。
微软表示，开发者想要使用 TensorFlow-DirectML 相当容易，因为它可以通过运行 pip install tensorflow-directml 命令在 Python 环境下安装。之后，它将自动与你现有的训练模型的脚本集成。除了鼓励开发者使用，微软还将直接与学生和专业人士合作，在这个过程中也会加强 GPU 调度和内存管理机制。
微软与 AMD、高通、英伟达和英特尔等供应商合作，确保在 Windows 10 和 WSL 之间提供流畅的体验，以便在支持 DirectX 12 的 GPU 上加速训练。就目前而言，TensorFlow-DirectML 的系统要求如下：
Windows 10
- Windows 10 1709，64位（Build 16299 或更高）；
- Python x86-64 3.5、3.6 或 3.7；
- 以下支持的 GPU 之一：
- AMD Radeon R5/R7/R9 2xx 系列或更新版本；
- 英特尔 HD Graphics 5xx 或更新版本；
- 英伟达 GeForce GTX 9xx 系列 GPU 或更新版本；
Windows Subsystem for Linux
- Windows 10 Insider Preview，64 位（Build 20150 或更高）；
- Python x86-64 3.5、3.6 或 3.7；
- 以下支持的 GPU 之一：
- AMD Radeon R5/R7/R9 2xx 系列或更新版本，以及 20.20.01.05 WSL 驱动；
- 英特尔 HD Graphics 6xx 或更新版本，以及 28.20.100.8322 WSL 驱动；
- 英伟达 GeForce GTX 9xx 系列或更新版本，以及 460.20 WSL 驱动；
注意：目前暂不支持 Python 3.8 及更高版本。要使用官方 PyPi 软件包，需要使用 CPython 解释器。NumPy 1.19.4 需要 KB4598291 才能在 Windows 上正常工作。
在未来，微软保证将继续进一步优化该项目，以及提供有针对性的支持。开发者可以在 GitHub 上查看并进一步了解这个开源项目。
