TensorFlow-DirectML TensorFlow 的 GPU 范围扩展开源项目

TensorFlow-DirectML是Tensorflow的一个分支，利用DirectML在Windows10和WSL上提供跨供应商的硬件加速。具有DirectML的TensorFlow支持在多种DirectX12兼容硬件上训练和推断复杂的机器学习模型。

TensorFlow-DirectML扩展了TensorFlow的范围，DirectML的加入使其超越了原有的GPU支持。它不仅扩展了TensorFlow在Windows上的GPU覆盖范围，而且还适用于WSL，用户可以使用任何DirectX12GPU在Windows或WSL环境中运行或训练其TensorFlow模型。