python - Keras 未检测到 GPU，但 TensorFlow 检测到

Question

我试图让我的 keras 后端工作，但它一直告诉我没有 GPU，但我的 tensorflow 却说不然。

tensorflow

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

输出

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 7600201089948511065
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 9640434864482537344
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 15836333527683385212
physical_device_desc: "device: XLA_GPU device"
]

喀拉斯

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

输出

[]

Tensorflow-gpu 可以工作，但 keras 由于某种原因看不到

Cuda已安装

tensorflow-gpu = 1.14.0
keras = 2.0.8
python = 3.6

Answer 1

在此之前您需要确认几件事，

您使用的是哪种 GPU？只有 NVIDIA 可以工作。你是否安装了tensorflow-gpu(keras使用它作为后端)？你安装了CUDA吗？

由于tensorflow正在检测GPU，那么它要么是上述几点的问题，要么您可以尝试升级各种软件包。

最后，如果没有其他办法，我建议使用tensorflow2中的高级API，它实际上与keras相同。

希望这有帮助。