IT资讯 微软 TensorFlow-DirectML 结束预览,正式发布

thesshell · 2021-09-12 09:30:07 · 热度: 18
去年 6 月,微软宣布它正在与硬件供应商合作一个名为 TensorFlow-DirectML 的项目,该项目可以在 Windows Subsystem for Linux(WSL)上提供 GPU 加速的机器学习(ML)模型的训练。经过了一年时间的预览,该项目如今已结束了预览,发布了正式版本。
微软 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 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 上查看并进一步了解这个开源项目。

猜你喜欢:
暂无回复。
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册