1. XenForo 1.5.14 中文版——支持中文搜索!现已发布!查看详情
  2. Xenforo 爱好者讨论群:215909318 XenForo专区

科技 TensorFlow-DirectML开放提供:在Windows 10/WSL上训练ML模型

Discussion in '新闻聚焦' started by 漂亮的石头, 2021-09-10.

  1. 漂亮的石头

    漂亮的石头 版主 Staff Member

    Joined:
    2012-02-10
    Messages:
    488,439
    Likes Received:
    48
    1 年多以前,微软宣布和多家硬件厂商合作,在 Windows Subsystem for Linux(WSL)上对 GPU 加速的机器学习(ML)模型进行训练。微软在 2020 年 6 月放出了预览版。今天,这个开源的 GitHub 项目已经退出预览,正式开放提供。

    [​IMG]

    微软训练了 TensorFlow-DirectML,这是 TensorFlow 的一个分叉,利用 DirectML 为在 Windows 10 和 WSL 上训练 ML 模型提供跨厂商的硬件加速。微软表示,使用 TensorFlow-DirectML 相当容易,因为它可以通过运行"pip install tensorflow-directml"命令在Python环境下安装。之后,它将自动与你现有的训练模型的脚本集成。​

    微软表示,它直接与学生和专业人士合作,为他们的脚本提供覆盖,并优化批量归一化和卷积等操作。这个过程也涉及加强 GPU 调度和内存管理机制。微软与 Nvidia、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 或更新版本​

    → NVIDIA GeForce GTX 9xx 系列 GPU 或更新版本​

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

    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驱动程序​

    → NVIDIA GeForce GTX 9xx系列GPU或更新版本,以及460.20 WSL驱动​

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