1 年多以前,微软宣布和多家硬件厂商合作,在 Windows Subsystem for Linux(WSL)上对 GPU 加速的机器学习(ML)模型进行训练。微软在 2020 年 6 月放出了预览版。 今天, 这个开源的 GitHub 项目已经退出预览 ,正式开放提供。 微软训练了 TensorFlow-DirectML,这是 TensorFlow 的一个分叉,利用 DirectML 为在 Windows 10 和 WSL 上训练 ML 模型提供跨厂商的硬件加速。微软表示,使用 TensorFlow-DirectML 相当容易,因为它可以通过运行"pip install tensorflow-directml"命令在Python环境下安装。之后,它将自动与你现有的训练模型的脚本集成。 微软表示,它直接与学生和专业人士合作,为他们的脚本提供覆盖,并优化批量归一化和卷积等操作。这个过程也涉及加强 GPU 调度和内存管理机制。微软与 Nvidia、 AMD 和 英特尔 等供应商合作,确保在 Windows 10 和 WSL 之间提供流畅的体验,以便在支持 DirectX 12 的 GPU 上加速培训。 【来源:cnBeta.COM】