TensorFlow-in-nvidia-docker-failed-call-to-cuInit-CUDA-ERROR-UNKNOWN

本文介绍了一种在Ubuntu上运行TensorFlow时遇到的CUDA错误“failed call to cuInit: CUDA_ERROR_UNKNOWN”的解决方法。通过调整系统配置文件和清理残留的NVIDIA驱动程序选项来修复该问题,使GPU支持恢复正常。

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

(转载请注明作者和出处:https://yangningbocn.github.io 未经允许请勿用于商业用途)

背景:
Ubuntu
TensorFlow


问题描述:
出现错误:
E TensorFlow in nvidia-docker: failed call to cuInit: CUDA_ERROR_UNKNOWN
然后就用CPU在跑


解决方法:

sudo update-alternatives --config x86_64-linux-gnu_gl_conf
9676334-cb51284b422ea6c5
解决之前

选择 /usr/lib/x86_64-linux-gnu/mesa/ld.so.conf
就好了
作为一个强迫症,你以为这样就完了吗,不!
我要把其他三个删掉
这个命令是用管理员的权限打开文件浏览器,然后删除那些不用的三个选项。这三个选项是我之前安装NVIDIA驱动后并卸载的残留

sudo nautilus 
9676334-5825a7a5f807e368
解决之后的图片

在/usr/lib/下 ctrl+f 输入nvidia 然后就能找到那些不用的选项删除就OK了。

/home/book/anaconda3/envs/yolov5face/lib/python3.9/site-packages/torch/serialization.py:1580: SourceChangeWarning: source code of class 'torch.nn.modules.container.Sequential' has changed. Saved a reverse patch to Sequential.patch. Run `patch -p0 < Sequential.patch` to revert your changes. warnings.warn(msg, SourceChangeWarning) /home/book/anaconda3/envs/yolov5face/lib/python3.9/site-packages/torch/serialization.py:1580: SourceChangeWarning: source code of class 'torch.nn.modules.conv.Conv2d' has changed. Saved a reverse patch to Conv2d.patch. Run `patch -p0 < Conv2d.patch` to revert your changes. warnings.warn(msg, SourceChangeWarning) /home/book/anaconda3/envs/yolov5face/lib/python3.9/site-packages/torch/serialization.py:1580: SourceChangeWarning: source code of class 'torch.nn.modules.batchnorm.BatchNorm2d' has changed. Saved a reverse patch to BatchNorm2d.patch. Run `patch -p0 < BatchNorm2d.patch` to revert your changes. warnings.warn(msg, SourceChangeWarning) /home/book/anaconda3/envs/yolov5face/lib/python3.9/site-packages/torch/serialization.py:1580: SourceChangeWarning: source code of class 'torch.nn.modules.activation.LeakyReLU' has changed. Saved a reverse patch to LeakyReLU.patch. Run `patch -p0 < LeakyReLU.patch` to revert your changes. warnings.warn(msg, SourceChangeWarning) /home/book/anaconda3/envs/yolov5face/lib/python3.9/site-packages/torch/serialization.py:1580: SourceChangeWarning: source code of class 'torch.nn.modules.container.ModuleList' has changed. Saved a reverse patch to ModuleList.patch. Run `patch -p0 < ModuleList.patch` to revert your changes. warnings.warn(msg, SourceChangeWarning) /home/book/anaconda3/envs/yolov5face/lib/python3.9/site-packages/torch/serialization.py:1580: SourceChangeWarning: source code of class 'torch.nn.modules.pooling.MaxPool2d' has changed. Saved a reverse patch to MaxPool2d.patch. Run `patch -p0 < MaxPool2d.patch` to revert your changes. warnings.warn(msg, SourceChangeWarning) /home/book/anaconda3/envs/yolov5face/lib/python3.9/site-packages/torch/serialization.py:1580: SourceChangeWarning: source code of class 'torch.nn.modules.upsampling.Upsample' has changed. Saved a reverse patch to Upsample.patch. Run `patch -p0 < Upsample.patch` to revert your changes. warnings.warn(msg, SourceChangeWarning) 2025-03-20 04:45:15.598387: E external/local_xla/xla/stream_executor/cuda/cuda_platform.cc:51] failed call to cuInit: INTERNAL: CUDA error: Failed call to cuInit: UNKNOWN ERROR (303)
03-21
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值