我正在研究使用keras和tensorflow后端的master的项目。我有intel(r)高清图形520,所以我无法使用tensorflow-gpu。 cpu版本工作正常。有没有办法使用tensorflow-gpu与intel(r)高清图形520?
答案 0 :(得分:5)
Tensorflow GPU支持需要安装Nvidia Cuda
和CuDNN
个软件包。对于GPU
加速培训,您需要专门的GPU
。 Intel
板载图片无法用于此目的。
您可以查看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 的其他讨论。