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

Posted

技术标签:

【中文标题】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: 无效语法

以上是关于Python:如何获取 GPU 的数量 [重复]的主要内容,如果未能解决你的问题,请参考以下文章