解决 Failed to load library libonnxruntime_providers_cuda.so with error: libcublasLt.so.12 问题

遇到问题

使用 onnxruntime-gpu==1.19.2 推理模型时遇到问题

[E:onnxruntime:Default, provider_bridge_ort.cc:1992 TryGetProviderInfo_CUDA] /onnxruntime_src/onnxruntime/core/session/provider_bridge_ort.cc:1637 onnxruntime::Provider& onnxruntime::ProviderLibrary::Get() [ONNXRuntimeError] : 1 : FAIL : Failed to load library libonnxruntime_providers_cuda.so with error: libcublasLt.so.12: cannot open shared object file: No such file or directory

另参考环境中 CUDA==11.2 大概可知是 onnxruntime-gpu 要求的 CUDA-12 与系统的 CUDA-11 版本不符导致。

查找方案

找到 github issue,确认了猜想:
https://github.com/microsoft/onnxruntime/issues/21684#issuecomment-2277205069

同时提供了 onnxruntime-gpuCUDA 对应版本的参考链接:
https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html

以下是不完整截图

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
注:CUDA 是大版本内互相兼容的,所以任意支持 CUDA-11 版本的 onnxruntime-gpu 都支持任意 CUDA-11 版本的环境。

解决问题

直接安装支持 CUDA-11 的 onnxruntime-gpu

python -m pip install onnxruntime-gpu==1.11

重新运行代码,顺利结束。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值