这几天在搭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!