tensorflow failed call to cuInit: CUDA_ERROR_UNKNOWN

最新推荐文章于 2025-02-24 20:47:11 发布

holomain

最新推荐文章于 2025-02-24 20:47:11 发布

阅读量3.7k

点赞数 7

CC 4.0 BY-SA版权

分类专栏： TensorFlow keras cuda error nvidia

kill yak-shaving

本文链接：https://blog.youkuaiyun.com/qq_27882063/article/details/82873355

TensorFlow 同时被 3 个专栏收录

12 篇文章

订阅专栏

keras

4 篇文章

订阅专栏

nvidia

2 篇文章

订阅专栏

本文针对使用NVIDIA GPU运行Keras代码时遇到的“tensorflow failed call to cuInit: CUDA_ERROR_UNKNOWN”错误，提供了详细的解决方案。包括安装nvidia-modprobe、设置CUDA_CACHE_PATH环境变量、启动nvidia-cuda-mps-server服务及重启计算机等步骤。文章还引用了多个权威社区的讨论链接，为读者提供更全面的故障排查视角。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用nvida GPU 跑keras 代码时报了如下错误：

“ tensorflow failed call to cuInit: CUDA_ERROR_UNKNOWN ”

软件版本：cuda 9.0
          tensorflow1.8
		  cudnn7-devel
		  ubuntu16.04

分别尝试了如下解决方法：
因为；nvidia-smi 和 nvcc -V 这两个命令均能返回正确结果。
所以做了如下尝试：

	01. sudo apt-get install nvidia-modprobe
	02. export CUDA_CACHE_PATH=/tmp/nvidia
	03. nvidia-cuda-mps-server
	04. reboot 
总有一种适合你。

过程中主要参考了如下大佬的讨论
https://github.com/tensorflow/tensorflow/issues/394

https://stackoverflow.com/questions/43992230/tensorflow-in-nvidia-docker-failed-call-to-cuinit-cuda-error-unknown/51120389#51120389

https://blog.youkuaiyun.com/jjj96321/article/details/76473017