TensorFlow-gpu 似乎没有看到我的 gpu

Posted 2023-03-12

【中文标题】TensorFlow-gpu 似乎没有看到我的 gpu

【英文标题】：Tensorflow-gpu doesn't seem to see my gpu

【发布时间】：2020-04-20 19:42:59

【问题描述】：

为了兼容性，我正在使用Which TensorFlow and CUDA version combinations are compatible? 中提到的 TensorFlow-GPU 1.12.0 版本、CUDA 9 版、CUDNN 7.0.5 版、bazel 0.15 版和 python 3.5.2 版。机器有Nvidia驱动384.130。

但是通过运行以下命令，Tensorflow 没有检测到我机器的 GPU

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

输出如下：

[

名称：“/设备：CPU：0” 设备类型：“CPU” memory_limit: 268435456

地点

化身：13408836213255819255 , 名称: "/device:XLA_CPU:0" 设备类型：“XLA_CPU” memory_limit: 17179869184

地点

化身：17981738971591465658 physical_device_desc：“设备：XLA_CPU 设备” ]

我尝试了以下链接中的解决方案

1-Tensorflow doesn't seem to see my gpu

2-list_local_device tensorflow does not detect gpu

【参考方案1】：

此解决方案适用于我的情况。

上述配置中CUDNN版本的问题。我将 CUDA 版本 9 的 CUDNN 版本更新为 7.1.4，我的代码开始使用 GPU。