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

Question

我正在嘗試在GPU上訓練Keras模型，Tensorflow作為后端。

我根據https://www.tensorflow.org/install/install_windows設置了所有內容。 這是我的設置：

• 我在virtualenv環境中使用Jupyter筆記本。
• 當前的virtualenv環境安裝了tensorflow-gpu 。
• 我安裝了CUDA 9.1和CUDA 9.1的cudaDNN。
• 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。

希望它在某種程度上有所幫助。