onnxruntime-gpu报错问题_无法定位onnxruntime-gpu

作者：繁依Fanyi0 | 2024-02-27 03:06:23

踩

无法定位onnxruntime-gpu

/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推理时，总是使用cpu；使用gpu时报错

onnxruntime版本不对
onnxruntime有cpu和gpu版本，分别对应
pip install onnxruntime 和 pip install onnxruntime-gpu
解决上一步后还有报错
pytorch官方的命令中，使用pip安装不会安装cudatoolkit，需要手动安装
conda install cudatoolkit

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/繁依Fanyi0/article/detail/149540