Tensorflow 未检测到 CUDA 设备

Posted 2023-03-16

技术标签:

【中文标题】Tensorflow 未检测到 CUDA 设备【英文标题】：Tensorflow not detecting CUDA device 【发布时间】：2021-11-09 18:11:04 【问题描述】：

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

使用 Tensorflow 运行一些代码时，我收到错误：

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 版本和驱动程序，但没有任何改变。

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

【问题讨论】：

我认为 Tensorflow2.x 不是为 CUDA >=10.2 预构建的。您可以从源代码构建它。 tensorflow.org/install/source 也许这会有所帮助***.com/questions/68237724/… 【参考方案1】：

您需要安装使用相同 CUDA 环境构建的软件包以确保兼容性。 Tensorflow 2.5 与 CUDA 11.2 兼容。

看看Tested build configuration

【讨论】：

【参考方案2】：

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

【讨论】：

以上是关于Tensorflow 未检测到 CUDA 设备的主要内容，如果未能解决你的问题，请参考以下文章