顾名思义,TensorFlow-DirectML 是 TensorFlow 的一个分叉,它既包括了 Google 的 TensorFlow 机器学习平台,也包括了微软的 DirectML API。开发者能够利用 TensorFlow-DirectML 为在 Windows 10 和 WSL 上训练 ML 模型获得跨厂商的硬件加速。
微软表示,开发者想要使用 TensorFlow-DirectML 相当容易,因为它可以通过运行 pip install tensorflow-directml 命令在 Python 环境下安装。之后,它将自动与你现有的训练模型的脚本集成。除了鼓励开发者使用,微软还将直接与学生和专业人士合作,在这个过程中也会加强 GPU 调度和内存管理机制。
pip install tensorflow-directml
微软与 AMD、高通、英伟达和英特尔等供应商合作,确保在 Windows 10 和 WSL 之间提供流畅的体验,以便在支持 DirectX 12 的 GPU 上加速训练。就目前而言,TensorFlow-DirectML 的系统要求如下:
注意:目前暂不支持 Python 3.8 及更高版本。要使用官方 PyPi 软件包,需要使用 CPython 解释器。NumPy 1.19.4 需要 KB4598291 才能在 Windows 上正常工作。
在未来,微软保证将继续进一步优化该项目,以及提供有针对性的支持。开发者可以在 GitHub 上查看并进一步了解这个开源项目。
评论删除后,数据将无法恢复
微软 TensorFlow-DirectML 结束预览,正式发布
顾名思义,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 Subsystem for Linux
注意:目前暂不支持 Python 3.8 及更高版本。要使用官方 PyPi 软件包,需要使用 CPython 解释器。NumPy 1.19.4 需要 KB4598291 才能在 Windows 上正常工作。
在未来,微软保证将继续进一步优化该项目,以及提供有针对性的支持。开发者可以在 GitHub 上查看并进一步了解这个开源项目。