我已经在我的环境中安装了onnxruntime gpu库
pip install onnxruntime-gpu==1.2.0
nvcc --version output
Cuda compilation tools, release 10.1, V10.1.105
>>> import onnxruntime
C:\Users\abgangwa\AppData\Local\Continuum\anaconda3\envs\onnx_gpu\lib\site-packages\onnxruntime\capi\_pybind_state.py:13: UserWarning: Cannot load onnxruntime.capi. Error: 'DLL load failed: The specified module could not be found.'
warnings.warn("Cannot load onnxruntime.capi. Error: '{0}'".format(str(e)))
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "C:\Users\abgangwa\AppData\Local\Continuum\anaconda3\envs\onnx_gpu\lib\site-packages\onnxruntime\__init__.py", line 12, in <module>
from onnxruntime.capi._pybind_state import get_all_providers, get_available_providers, get_device, RunOptions, SessionOptions, set_default_logger_severity, NodeArg, ModelMetadata, GraphOptimizationLevel, ExecutionMode
ImportError: cannot import name 'get_all_providers'
请帮助我解决此错误
尝试了现有的解决方案。它们都不起作用
发布于 2021-04-28 13:23:00
很可能CUDA dll不在路径中,所以在python加载onnxruntime库时找不到它。
此回复有一个链接,指向一篇关于修复此问题的文章。
https://github.com/microsoft/onnxruntime/issues/4196#issuecomment-808787338
https://stackoverflow.com/questions/66156292
复制相似问题