🛠️ 在线工具
📰 新闻动态

欢迎回来

登录你的 AI Flow 账号,继续创作之旅

还没有账号?立即注册

创建账号

加入 AI Flow,解锁全部 ComfyUI 教程与工作流

两次输入的密码不一致
验证码错误
已有账号?去登录

onnxruntime::TryGetProviderInfo_CUDA

[E:onnxruntime:Default, provider_bridge_ort.cc:2395 onnxruntime::TryGetProviderInfo_CUDA] N:\_work\1\s\onnxruntime\core\session\provider_bridge_ort.cc:1988 onnxruntime::ProviderLibrary::Get [ONNXRuntimeError] : 1 : FAIL : Error loading "D:\Miniconda3\envs\comfyuicu118\Lib\site-packages\onnxruntime\capi\onnxruntime_providers_cuda.dN0Rc歔剉!jWW0")s on "cublasLt64_13.dll" which is missing. (Error 126: "~b


解决方案
 

这个报错的意思是 ONNX Runtime 在尝试加载 CUDA 执行提供程序(CUDAExecutionProvider)时失败了,原因通常是 ‌onnxruntime-gpu 版本和本机的 CUDA、cuDNN 版本不匹配‌,或者相关的动态链接库(DLL /.so 文件)找不到。‌‌

核心排查方向

‌版本匹配是关键‌:onnxruntime-gpu 对 CUDA 和 cuDNN 版本有严格要求。你可以参考 ONNX Runtime 官方文档确认对应关系,常见组合为:

  • CUDA 12.x + cuDNN 9.x
  • CUDA 11.x + cuDNN 8.x ‌‌

‌操作步骤‌:

  1. ‌检查本机环境‌:通过 nvcc --version 查看 CUDA 版本,通过 cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR(Linux)或查看安装目录(Windows)确认 cuDNN 版本。
  2. ‌核对并安装匹配版本‌:根据你的 CUDA/cuDNN 版本,在官方文档找到对应的 onnxruntime-gpu 版本并安装。例如 CUDA 12.x 通常安装 onnxruntime-gpu==1.18.x 或更新的版本。不要直接用 pip install onnxruntime-gpu 安装默认最新版,建议指定版本号。
  3. ‌检查环境变量‌:确保 CUDA 和 cuDNN 的 bin 目录(Windows)或 lib 目录(Linux)已添加到系统 PATH 或 LD_LIBRARY_PATH 中。
  4. ‌避免版本冲突‌:确保环境中没有同时安装 onnxruntime(CPU版)和 onnxruntime-gpu,两者会冲突。‌‌

其他可能情况

  • ‌缺少特定库文件‌:报错可能具体指出缺少哪个文件,如 libcudnn.so.9、libcublasLt.so.12 等。这通常意味着对应库未安装或未加入路径,按提示补齐对应库文件即可。
  • ‌降级或升级 onnxruntime-gpu‌:如果当前版本不匹配,可以尝试降级或升级到与你的 CUDA 环境匹配的版本。有案例是将 1.20.1 降级到 1.18.0 后成功。‌‌

如果以上都排查无误但问题依旧,可以考虑使用 Docker 容器来隔离并标准化 CUDA 环境,避免本机环境干扰。‌‌