TensorFlow-DirectML 是 Tensorflow 的一个分支,利用 DirectML 在 Windows 10 和 WSL 上提供跨供应商的硬件加速。具有 DirectML 的 TensorFlow 支持在多种 DirectX 12 兼容硬件上训练和推断复杂的机器学习模型。
TensorFlow-DirectML 扩展了 TensorFlow 的范围,DirectML 的加入使其超越了原有的 GPU 支持。它不仅扩展了 TensorFlow 在 Windows 上的 GPU 覆盖范围,而且还适用于 WSL,用户可以使用任何 DirectX 12 GPU 在 Windows 或 WSL 环境中运行或训练其 TensorFlow 模型。