微软开源 TensorFlow-DirectML,为 WSL2 提供 GPU 支持
微软近日开源了TensorFlow-DirectML 项目,它是 Tensorflow 的一个分支,利用 DirectML 在 Windows 10 和 WSL 上提供硬件加速。 DirectML 是微软计划为 WSL2 提供的两种 GPU 计算功能之一。它本身能够通过广泛支持的硬件和驱动程序为常见的机器学习任务提供 GPU 加速,包括来自 AMD、Intel、NVIDIA 和 Qualcomm 等供应商的所有支持 DirectX 12 的 GPU。 今年六月份,WSL2 已达到了对 GPU 的初始支持,其中就包括对DirectML的支持。当时,微软发布了带有 DirectML 后端的 TensorFlow 预览包,而这次开源的TensorFlow-DirectML 是一个完整的Tensorflow 分支。 微软高级程序经理 Clarke Rahrig 等人表示,TensorFlow-DirectML 扩展了 TensorFlow 的范围,DirectML 的加入使其超越了原有的 GPU 支持。 “机器学习工作负载通常涉及大量计算,尤其是在训练模型时。诸如 GPU 之类的专用硬件通常用...

