无论如何使用tensorflow-gpu与intel(r)高清图形520?

时间:2017-11-20 19:40:29

标签: deep-learning keras tensorflow-gpu

我正在研究使用keras和tensorflow后端的master的项目。我有intel(r)高清图形520,所以我无法使用tensorflow-gpu。 cpu版本工作正常。有没有办法使用tensorflow-gpu与intel(r)高清图形520?

2 个答案:

答案 0 :(得分:5)

Tensorflow GPU支持需要安装Nvidia CudaCuDNN个软件包。对于GPU加速培训,您需要专门的GPUIntel板载图片无法用于此目的。

您可以查看tensorflow-gpu here

的完整要求

答案 1 :(得分:1)

我刚刚看到了 this page(2020 年 9 月发布)。因此,可以使用 DirectML 库在 DirectX 12 兼容的 GPU 上运行 Tensorflow。它被称为TensorFlow-DirectML

<块引用>

TensorFlow 是用于机器学习的端到端开源平台。此存储库是 tensorflow 的一个分支,它利用 DirectML 在 Windows 10 和适用于 Linux 的 Windows 子系统 (WSL) 上提供跨供应商硬​​件加速。 TensorFlow with DirectML 支持在各种 DirectX 12 兼容硬件上训练和推理复杂的机器学习模型。

还有一些关于 OpenCL here 的其他讨论。