Python:如何获取 GPU 数量

Question

我使用 Keras 和 Tensorflow 来复制 DQN。实际上我使用的是 2 个 GPU，但我想让我的代码在每台机器上都可执行，无论它有多少个 GPU。所以我的问题是:如何获取我的 PC 中存在的 GPU 数量？我检查了this但它给了我很多我不需要的信息。还有其他方法可以获取 GPU 数量吗？

Answer 1

如果您想知道有多少个 GPU 正在运行，tf.contrib.eager.num_gpus() 将为您提供答案。