TensorFlow-DirectMLTensorFlow 的 GPU 范围扩展

联合创作 · 2023-09-26 05:43

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 模型。

浏览 6
点赞
评论
收藏
分享

手机扫一扫分享

编辑
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑
举报