时间戳

首页 > 解决方案 > Tensorflow-gpu 没有检测到 GPU？

python-3.x - Tensorflow-gpu 没有检测到 GPU？

问题描述

我在 python 3.7.4 中与 Anaconda 一起安装了 tensorflow-gpu 版本 2.2.0，但我的代码始终在 CPU 上运行并且无法检测到我的 GPU。

physical_devices= tf.config.experimental.list_physical_devices('GPU')
print(len(physical_devices))
>>> 0

当我运行这个：

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: 6159543114609950707
, name: "/device:XLA_CPU:0"
device_type: "XLA_CPU"
memory_limit: 17179869184
locality {
}
incarnation: 4043157374881641271
physical_device_desc: "device: XLA_CPU device"
, name: "/device:XLA_GPU:0"
device_type: "XLA_GPU"
memory_limit: 17179869184
locality {
}
incarnation: 1519736160969870434
physical_device_desc: "device: XLA_GPU device"
]

XLA_GPU:0 对应于在 CPU 上运行的集成显卡，因为我已经测试过在tf.device('/GPU:0'):下运行我的代码：并且任务管理器仅显示 CPU 使用率。

我检查了我的 NVIDIA 控制面板 -> 系统信息 -> 组件，在 3D 设置下，我将 NVCUDA64.dll 描述为 NVIDIA CUDA 11.0.208 驱动程序。

到目前为止，我还没有尝试重新安装 tensorflow-gpu，期待之前可能还有其他事情要做。

标签： python-3.xtensorflowgpu

解决方案

我自己的答案。最后它起作用了：

我必须为该版本下载 CUDA 10.1 和 cuDNN 7.6.5。

感谢@Gerry P 和@Richard X

推荐阅读