python - Python：如何获取 GPU 的数量

Question

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

标签： pythontensorflowkerasgpu

Answer 1

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