gpt4 book ai didi

Python:如何获取 GPU 数量

转载 作者:太空宇宙 更新时间:2023-11-03 21:02:22 24 4
gpt4 key购买 nike

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

最佳答案

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

关于Python:如何获取 GPU 数量,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55643308/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com