Python：如何获取 GPU 的数量 [重复]

Posted 2023-02-23

【中文标题】Python：如何获取 GPU 的数量 [重复]

【英文标题】：Python: how to get the number of GPUs [duplicate]

【发布时间】：2019-09-02 17:27:59

【问题描述】：

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

【问题讨论】：

你链接的答案字面上有答案，他们列举了 GPU，你只需要得到那个列表的长度。

我是这么想的，但我想知道是否有其他方法。这样，您的系统中就有 gpus + cpus 的数量

不，他们专门过滤以仅获取 GPU。

【参考方案1】：

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

【讨论】：

当我尝试使用它时，它给了我以下错误：文件“”，第 28 行 def single_image_random_dot_stereograms(depth_values, hidden_​​surface_removal=True,convergence_dots_size=8,dots_per_inch=72, eye_separation= 2,5, mu=0,333299994, normalize=True, normalize_max=-100, normalize_min=100, border_level=0, number_colors=256, output_image_shape=[1024, 768, 1], output_data_window=[1022, 757], 名称=None): ^ SyntaxError: 无效语法