去年 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 上查看并进一步了解这个开源项目。
暂无回复。