python-3.x - 查看 Keras 和 Tensorflow 是否使用 GPU

Question

我在运行 Windows 10 并有两个显卡的 PC 上安装了 tensorflow-gpu：

我正在尝试复制 Tensorflow 文档：

但是，我没有得到文档中描述的内容：

如何确定 Tensorflow 是否使用了我的 GPU？我相信我已正确遵循所有设置说明，但我该如何验证呢？

我了解 Tensorflow 默认使用可用的 GPU。

更新

@鲨鱼：

这就是我按照你的指示得到的，如果我理解得很好的话。我应该推断什么以及接下来应该做什么？

@马诺吉

这是我从命令提示符调用的 python shell 运行代码时得到的：

消息的含义和含义是什么？

标签： python-3.xtensorflowkerasgpu

Answer 1

您必须安装支持 gpu 的 tensorflow，通常pip install tensorflow只支持 cpu。在这里你可以找到 gpu tensorflow 安装指南。之后，当您在终端中运行 log_device_placement 示例时，您应该获得有关 gpu 使用情况的信息。