模块keras.backend没有属性“tensorflow_backend”
模块keras.backend没有属性“tensorflow_backend”
提问于 2020-07-23 08:04:11
当我运行我的代码时，我试图使用张量流作为后端，并执行以下操作：
import tensorflow
KERAS_BACKEND=tensorflow
import keras
from keras import backend as K
K.tensorflow_backend._get_available_gpus()我得到以下错误：
K.tensorflow_backend._get_available_gpus()
AttributeError: module 'keras.backend' has no attribute 'tensorflow_backend'但是，当我在木星笔记本上运行相同的代码时，它可以工作，并给出以下输出：
['/job:localhost/replica:0/task:0/device:GPU:0']这个输出意味着安装了张量流。当我通过终端运行相同的脚本时，我不明白为什么会得到上面提到的错误。在这两种情况下，我所处的环境是相同的。我会很感激你的见解。
回答 1
Stack Overflow用户
回答已采纳
发布于 2020-09-23 16:11:29
从您正在尝试理解的代码中，Tensorflow是否已安装并使用GPU？
在Tensorflow 2.x中，可以通过以下方式知道相同的情况
tf.config.list_physical_devices('GPU') 输出：
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]下面还将返回GPU设备的名称。
import tensorflow as tf
tf.test.gpu_device_name()如果安装了包的non-GPU版本，该函数也将返回False。使用tf.test.is_built_with_cuda验证TensorFlow是否具有CUDA支持。
注:从TF-2.0开始，Keras与Tensorflow集成为tf.keras，请使用此模块tf.keras.backend
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/63049620
复制相关文章
相似问题
腾讯云开发者
