Warning: file_get_contents(/data/phpspider/zhask/data//catemap/2/python/331.json): failed to open stream: No such file or directory in /data/phpspider/zhask/libs/function.php on line 167

Warning: Invalid argument supplied for foreach() in /data/phpspider/zhask/libs/tag.function.php on line 1116

Notice: Undefined index: in /data/phpspider/zhask/libs/function.php on line 180

Warning: array_chunk() expects parameter 1 to be array, null given in /data/phpspider/zhask/libs/function.php on line 181
Python:如何获取GPU的数量_Python_Tensorflow_Keras_Gpu - Fatal编程技术网

Python:如何获取GPU的数量

Python:如何获取GPU的数量,python,tensorflow,keras,gpu,Python,Tensorflow,Keras,Gpu,我用Keras和Tensorflow来复制一个DQN。实际上我使用的是2个GPU,但是我想让我的代码在每台机器上都可以执行,不管它有多少个GPU。所以我的问题是:我如何才能得到我的电脑中存在的GPU的数量?我查过了,但它给了我很多我不需要的信息。有没有其他方法可以获取GPU的数量?如果您想知道有多少GPU正在运行tf.contrib.eager.num\u GPUs()将为您提供答案 你链接的答案实际上就是答案,他们列举了GPU,你只需要得到列表的长度。这就是我的想法,但我想知道是否还有其他方法

我用Keras和Tensorflow来复制一个DQN。实际上我使用的是2个GPU,但是我想让我的代码在每台机器上都可以执行,不管它有多少个GPU。所以我的问题是:我如何才能得到我的电脑中存在的GPU的数量?我查过了,但它给了我很多我不需要的信息。有没有其他方法可以获取GPU的数量?

如果您想知道有多少GPU正在运行
tf.contrib.eager.num\u GPUs()
将为您提供答案

你链接的答案实际上就是答案,他们列举了GPU,你只需要得到列表的长度。这就是我的想法,但我想知道是否还有其他方法。这样,您的系统中就有了GPU+CPU的数量否,它们专门过滤以仅获取GPU。当我尝试使用它时,它会给我以下错误:文件“”,第28行def single_image_random_dot_立体图(深度值、隐藏表面移除值=真、收敛点大小=8、每英寸点数=72、眼睛间距=2,5、mu=033329994、规格化值=真、规格化最大值=-100、规格化最小值=100、边框级别=0、颜色数=256、输出图像形状=[1024768,1],输出数据窗口=[1022757],名称=无):^SyntaxError:无效语法