onnxruntime-gpu飘红报错怎么解决?

作者在使用insightface时遇到GPU报错,发现是onnxruntime-gpu版本问题。通过升级到1.17.1版本(通过pipinstall--upgradeonnxruntime-gpu--extra-index-urlhttps://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/),解决了CUDA版本不匹配导致的加载失败问题,建议对相似困境的开发者参考此解决方案。

这几天在搭comfyui环境的时候遇到一个棘手的问题,使用insightface的时候,GPU报错,只能用CPU,虽然能够运行,但也降低了运行速度。

报错就是大概这样的,在控制台中是明显的红色:

/onnxruntime_src/onnxruntime/core/session/provider_bridge_ort.cc:1193 onnxruntime::Provider& onnxruntime::ProviderLibrary::Get() [ONNXRuntimeError] : 1 : FAIL : Failed to load library libonnxruntime_providers_cuda.so with error: libcudart.so.11.0: cannot open shared object file: No such file or directory

报错有少许不同,就类似这种,不用查也知道是onnxruntime-gpu版本的问题,但一直不知道该是哪个版本,今天看到个解决方法,终于有用了。

pip install --upgrade onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/

这样安装的是1.17.1的版本。

原本我是1.17.0的版本,cuda的话应该比较高,尝试了1.16多的版本也不行,如果有这个困扰的小伙伴,可以用这个方式,去找下比较搭配的版本。

祝大家成功赶走bug!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值