Tensorflow-gpu 似乎沒有看到我的 gpu

Question

我正在使用 TensorFlow-GPU 1.12.0 版本、CUDA 版本 9、CUDNN 版本 7.0.5、bazel 版本 0.15 和 python 版本 3.5.2，如哪個 TensorFlow 和 CUDA 版本組合兼容？ 為了兼容性。 該機器具有 Nvidia 驅動程序 384.130。

但是通過運行以下命令，Tensorflow 未檢測到我機器的 GPU

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

輸出如下：

[

名稱：“/設備：CPU：0”設備類型：“CPU”內存限制：268435456

地點{}

化身：13408836213255819255，名稱：“/device:XLA_CPU:0” device_type：“XLA_CPU” memory_limit：17179869184

地點{}

化身：17981738971591465658 physical_device_desc：“設備：XLA_CPU設備”]

我嘗試了以下鏈接中的解決方案

1- Tensorflow 似乎沒有看到我的 GPU

2- list_local_device tensorflow 不檢測 gpu

Answer 1

這個解決方案適用於我的情況。

上面配置中CUDNN版本的問題。 我將 CUDNN 版本更新為 CUDA 版本 9 的 7.1.4，我的代碼開始使用 GPU。