Python：如何获取GPU数量

Question

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

Answer 1

如果您想知道正在运行tf.contrib.eager.num_gpus()的GPU数量，则会为您提供答案。