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。