onnxruntime-gpu报错问题
时间:2024-05-02 12:55:46 来源:网络cs 作者:利杜鹃 栏目:卖家故事 阅读:
阅读本书更多章节>>>>
onnxruntime有cpu和gpu版本,分别对应
pytorch官方的命令中,使用pip安装不会安装cudatoolkit,需要手动安装
/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
阅读本书更多章节>>>>
本文链接:https://www.kjpai.cn/gushi/2024-05-02/164253.html,文章来源:网络cs,作者:利杜鹃,版权归作者所有,如需转载请注明来源和作者,否则将追究法律责任!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容,一经查实,本站将立刻删除。
上一篇:C++算法:前缀和、前缀乘积、前缀异或的原理、源码及测试用例
下一篇:返回列表