文章/答案/技术大牛

发布

社区首页 >问答首页 >Tensorflow未检测到CUDA装置

问Tensorflow未检测到CUDA装置
EN

Stack Overflow用户

提问于 2021-09-14 14:39:48

回答 2查看 1.2K关注 0票数 1

我对Tensorflow有一些问题，这似乎没有检测到我的GPU。

当使用Tensorflow运行一些代码时，我得到了错误

tensorflow/stream_executor/cuda/cuda_driver.cc:328]

对cuInit的调用失败: CUDA_ERROR_NO_DEVICE:没有检测到具有数据自动化功能的设备

这是我的配置

Nvidia GeForce RTX 3080 Ti
Ubuntu 18.04
CUDA 11.4，司机470.57.02
Tensorflow 2.5

我的GPU被很好地检测到(用nvidia-smi检查过)，tf.test.is_gpu_available()返回True。我试着降低CUDA版本和驱动程序的等级，但是没有什么改变。

有人对如何解决这个问题有什么建议吗？非常感谢!

python

tensorflow

gpu

nvidia

回答 2

Stack Overflow用户

发布于 2021-10-01 07:22:49

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

看看经过测试的构建配置

票数 1

Stack Overflow用户

发布于 2021-10-01 16:49:22

此问题是由于TensorFlow 2.5兼容。所以，只要把你的数据自动化系统降级到11.2。https://developer.nvidia.com/cuda-11.2.0-download-archive

票数 1

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/69179815

复制

相似问题

问Tensorflow未检测到CUDA装置
EN

回答 2

Stack Overflow用户

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问Tensorflow未检测到CUDA装置EN

回答 2

Stack Overflow用户

Stack Overflow用户

社区

活动

圈层

关于

腾讯云开发者

热门产品

热门推荐

更多推荐

问Tensorflow未检测到CUDA装置
EN