我在调用onnxruntime-GPU的时候遇到了问题
我创建了一个虚拟环境,python3.7
我的cuda版本为10.2,onnx为1.8.0,onnxruntime-gpu为1.6.0
当我尝试import onnxruntime的时候出现报错
>>> import onnxruntime
C:\Users\ASUS\anaconda3\envs\yolo3\lib\site-packages\onnxruntime\capi\_pybind_state.py:14: UserWarning: Cannot load onnxruntime.capi. Error: 'DLL load failed: 找不到指定的模块。'.
warnings.warn("Cannot load onnxruntime.capi. Error: '{0}'.".format(str(e)))
Traceback (most recent call last):
File "" , line 1, in <module>
File "C:\Users\ASUS\anaconda3\envs\yolo3\lib\site-packages\onnxruntime\__init__.py", line 13, in <module>
from onnxruntime.capi._pybind_state import get_all_providers, get_available_providers, get_device, set_seed, \
ImportError: cannot import name 'get_all_providers' from 'onnxruntime.capi._pybind_state' (C:\Users\ASUS\anaconda3\envs\yolo3\lib\site-packages\onnxruntime\capi\_pybind_state.py)
请问该如何解决呢,感觉相关信息很少。大部分回答都是说cuda版本不对,但我的应该没问题
不知道你这个问题是否已经解决, 如果还没有解决的话:建议不要用conda环境试一下