Ubuntu16下TensorFlow2.5加载模型时报libcudart.so.11.0缺失及cuInit错误求助
问题成因
- CUDA相关警告:属于非致命提示,出现原因是本地未安装适配TensorFlow 2.5.0的NVIDIA显卡驱动、CUDA 11.0运行库,TensorFlow未检测到可用GPU,将自动切换为CPU模式运行,不影响代码正常执行。
- 核心运行问题:代码执行后无推理任务运行(进度条显示
0it [00:00, ?it/s]),是因为glob匹配的测试文件路径未找到任何有效文件,test_images为空列表,后续循环逻辑完全未执行。 - 潜在语法问题:你提供的代码中,模型加载行下方存在一个孤立的反引号
`,如果实际运行的代码也包含该符号,会直接触发语法错误终止程序运行。
修复方案
- 修复路径匹配问题
在test_images = sorted(glob("/mnt/DATA2To/projet/all/Souris/SOD/data/20210726_C321/EXVIVO/IRM/test-20210726_C321/*"))行后添加调试代码:
若输出为0,按如下顺序排查:print("匹配到的测试文件数量:", len(test_images))- 确认路径大小写完全匹配,Linux系统对路径大小写敏感
- 确认当前运行代码的用户拥有
/mnt/DATA2To对应目录的读取权限 - 确认测试文件确实存放在目标路径下,未被移动、删除或重命名
- 清理代码语法错误
删除模型加载行下方多余的孤立反引号,避免触发语法错误。 - 可选:消除CUDA警告
若不需要使用GPU运行,可在导入TensorFlow前添加代码屏蔽警告:
若需要使用GPU运行,安装适配TensorFlow 2.5.0的NVIDIA驱动(版本≥450.80.02)、CUDA 11.0、cuDNN 8.0即可。import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
内容的提问来源于stack exchange,提问作者boubekeur anis
相关产品推荐
相关产品推荐

