微软首次广泛发布 TensorFlow-DirectML

微软首次广泛发布 TensorFlow-DirectML
去年,微软一直在其 Windows 子系统 Linux (WSL) 服务上为机器学习模型开发 GPU 加速训练。经过最后一年的预览,微软现在将这个项目作为 TensorFlow-DirectML 启动。

顾名思义,TensorFlow-DirectML 是一个分支,包括谷歌的 TensorFlow 机器学习平台和微软的 DirectML API。

DirectML 为机器学习任务提供 GPU 加速工具。DirectML 在 AMD、高通、英伟达和英特尔芯片上的 DirectX 12 GPU 上跨硬件和驱动程序工作。借助 TensorFlow-DirectML,微软提供了一个硬件加速平台,可供多个供应商访问以训练机器学习模型。

微软表示,这个开源项目现在可以通过 GitHub 完全获得。

“今天,我们很高兴退出预览版并宣布我们的第一个通用 TensorFlow-DirectML 软件包!我们鼓励您使用 TensorFlow-DirectML,无论您是学生学习者还是专业开发机器学习模型以进行生产。请继续阅读有关我们如何投资以及下一步走向的更多详细信息。”

为了帮助发布 TensorFlow-DirectML 的最佳版本,微软征求了用户反馈,并与 AMD、英特尔和英伟达等合作伙伴合作。微软表示运行该服务需要满足以下要求。

要求

视窗 10

  • Windows 10 版本 1709,64 位(内部版本 16299 或更高版本)
  • Python x86-64 3.5、3.6 或 3.7
  • 以下支持的 GPU 之一:
    • AMD Radeon R5/R7/R9 2xx 系列或更新
    • Intel HD Graphics 5xx 或更新版本
    • NVIDIA GeForce GTX 9xx 系列 GPU 或更新

注意:当前不支持 Python 3.8 或更新版本。要使用官方 PyPi 包,需要 CPython 解释器。NumPy 1.19.4 需要 KB4598291 才能在 Windows 上正常工作。

适用于 Linux 的 Windows 子系统

  • 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 驱动程序
    • Intel HD Graphics 6xx 或更新版本,以及 28.20.100.8322 WSL 驱动程序
    • NVIDIA GeForce GTX 9xx 系列 GPU 或更新版本,以及 460.20 WSL 驱动程序

注意:当前不支持 Python 3.8 或更新版本。要使用官方 PyPi 包,需要 CPython 解释器。

新的 TensorFlow-DirectML 可在GitHub 上获得。

正文完