python - Keras 不使用 GPU - 如何排除故障？

Question

我正在尝试在 GPU 上训练 Keras 模型，并使用 Tensorflow 作为后端。

我已经根据 https://www.tensorflow.org/install/install_windows 设置了一切。这是我的设置:

• 我正在 virtualenv 环境中的 Jupyter 笔记本中工作。
• 当前 virtualenv 环境已安装 tensorflow-gpu。
• 我安装了 CUDA 9.1 和 cudaDNN for CUDA 9.1。
• cuDNN64_7.dll 位于可通过 PATH 变量访问的位置。
• 我的计算机上有 NVIDIA GeForce GTX 780，并且安装了最新驱动程序。

但是，Tensorflow 没有看到任何可用的 GPU:

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: 5275203639471190827
]

Keras 都没有:

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

[]

我该如何调试这个？如何找出问题出在哪里？

Answer 1

检查
nvcc-V
和

nvidia-smi

看看它是否显示我们的 GPU。

假设你的 cuda cudnn 和一切都检查完毕，你可能只需要
1.卸载keras
2.卸载tensorflow
3.卸载tensorflow-gpu
4.仅安装tensorflow-gpu pip install tensorflow-gpu==1.5.0
5. 立即安装 Keras。

我按照这些步骤操作，keras 现在使用 GPU。

希望能在一定程度上有所帮助。