当前位置: 首页 > 知识库问答 >
问题:

如何检查keras tensorflow后端是GPU还是CPU版本?[副本]

左康安
2023-03-14

我知道在安装tensorflow时,您可以安装GPU或CPU版本。如何检查安装了哪一个(我使用linux)。

如果安装了GPU版本,如果GPU不可用,它会自动在CPU上运行吗?如果GPU可用,是否需要设置特定的字段或值来确保它在GPU上运行?

共有2个答案

伯建安
2023-03-14

根据文件。

如果您在TensorFlow或CNTK后端上运行,如果检测到任何可用的GPU,您的代码将自动在GPU上运行。

你可以通过以下方式检查所有的设备都被tenorflow使用-

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

也正如这个回答中所建议的那样

import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

这将打印您的tensorflow是使用CPU还是GPU后端。如果您在jupyter笔记本中运行此命令,请查看启动笔记本的控制台。

如果你怀疑你是否安装了tensorflow gpu版本。你可以通过pip安装gpu版本。

pip安装tensorflow gpu

顾嘉德
2023-03-14

您还可以使用Keras后端功能进行检查:

from keras import backend as K
K.tensorflow_backend._get_available_gpus()

我在Keras上对此进行了测试(2.1.1)

 类似资料:
  • 问题内容: 运行keras脚本时,得到以下输出: 这是什么意思?我是否正在使用GPU或CPU版本的Tensorflow? 在安装keras之前,我正在使用Tensorflow的GPU版本。 还显示和没有什么像。 运行[此stackoverflow问题]中提到的命令,将得到以下信息: 问题答案: 您正在使用GPU版本。您可以列出可用的tensorflow设备(也请检查此问题): 编辑: 使用tens

  • 当我运行keras脚本时,我得到以下输出: 这是什么意思?我是否使用GPU或CPU版本的tenstorflow? 在安装keras之前,我使用的是tensorflow的GPU版本。 另外显示了,与完全不同。 运行[this stackoverflow question]中提到的命令,可以得到以下结果:

  • 如何检查名称为“session”的表是否存在?我如何知道我的查询是否有效? 编辑:这不是上述问题的重复,因为我的问题与Node.js javascript代码有关。不是SQL!我只想知道在res对象中查找什么,因为它似乎是未定义的...

  • 问题内容: 是否可以检查a 是正零(0.0)还是负零(-0.0)? 我已经将转换为,并检查第一个是否为,但是还有其他方法吗? 问题答案: 是的,除以它。是,但是是。通过简单的比较很容易找出是哪一个,因此您得到: (这假定只能是两个零之一)

  • 本文向大家介绍如何检查iOS程序是在前台还是在后台?,包括了如何检查iOS程序是在前台还是在后台?的使用技巧和注意事项,需要的朋友参考一下 作为iOS开发人员,了解应用程序处于前台还是后台很重要,我们需要处理多个事件,例如后台下载,应用程序进入前台时的事件。 在这里,我们将看到如何检查应用程序是在后台还是在前台。 我们将为此使用通知中心, 要了解更多信息,请参考Apple文档。  https://

  • 问题内容: 我有一个通过JSON.Stringify函数从对象转换的json字符串。 我想知道它是json字符串还是普通字符串。 是否有类似“ isJson()”的函数来检查它是否为json? 我想在使用本地存储(如下面的代码)时使用该函数。 先感谢您!! 问题答案: “简单”的方法是在失败时解析并返回未解析的字符串: