Windows 11
微软最新的操作系统

微软通过 TensorFlow-DirectML 在 Windows 10 和 WSL 上提供 GPU 加速的机器学习训练

微软宣布正在与硬件供应商合作,在 Windows Subsystem for Linux (WSL) 上提供机器学习 (ML) 模型的 GPU 加速训练。对此的预览于 2020 年 6 月开始。该计划的成果是 TensorFlow-DirectML,它是 TensorFlow 的一个分支,它利用 DirectML 为在 Windows 10 和 WSL 上训练 ML 模型提供跨供应商硬​​件加速。今天,这个开源的 GitHub 项目已经退出预览版并普遍可用

1588189631_ms_loves_linux_story

微软已经吹捧使用 TensorFlow-DirectML 非常容易,因为它可以通过运行命令“pip install tensorflow-directml”安装在 Python 环境中。之后,它将自动与您现有的用于训练模型的脚本集成。这家雷德蒙德科技巨头表示,它直接与学生和专业人士合作,为他们的脚本提供覆盖,并优化批处理规范化和卷积等运算符。这个过程也涉及增强 GPU 调度和内存管理机制。

微软与Nvidia等供应商合作、AMD 和英特尔,以确保在 Windows 10 和 WSL 上提供流畅的体验,以便在支持 DirectX 12 的 GPU 上加速训练。目前,TensorFlow-DirectML 的系统要求如下:

视窗 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 或更新版本
    • 英伟达 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 驱动程序
    • 英伟达 GeForce GTX 9xx 系列 GPU 或更新版本,以及 460.20 WSL 驱动程序

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

向前迈进,微软已保证它将继续在有针对性的运营商支持下进一步优化项目。您可以在此处查看 GitHub 上的开源计划

赞(0) 打赏
未经允许不得转载:Win 11系统之家 » 微软通过 TensorFlow-DirectML 在 Windows 10 和 WSL 上提供 GPU 加速的机器学习训练

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏