python - Tensorflow 未检测到 CUDA 设备

Question

我在使用 Tensorflow 时遇到了一些问题，似乎没有检测到我的 GPU。

使用 Tensorflow 运行一些代码时，出现错误：

张量流/stream_executor/cuda/cuda_driver.cc:328]

对 cuInit 的调用失败：CUDA_ERROR_NO_DEVICE：未检测到支持 CUDA 的设备

这是我的配置：

• 英伟达 GeForce RTX 3080 Ti
• Ubuntu 18.04
• CUDA 11.4，驱动程序 470.57.02
• 张量流 2.5

我的 GPU 被很好地检测到（用 进行了检查nvidia-smi）并tf.test.is_gpu_available() 返回 True。我尝试降级 CUDA 版本和驱动程序，但没有任何改变。

有人对如何解决这个问题有一些提示吗？非常感谢！

标签： pythontensorflowgpunvidia

Answer 1

由于 TensorFlow 2.5 兼容而出现此问题。因此，只需将您的 CUDA 降级（重新安装）到 11.2。 https://developer.nvidia.com/cuda-11.2.0-download-archive