报错如下:
Traceback (most recent call last):
File "test.py", line 7, in <module>
from utils.utils_inference import InterfaceOnnx, InterfaceTorch
File "F:\F_code\insightface_ir_train\utils\utils_inference.py", line 8, in <module>
import onnxruntime
File "D:\anaconda\envs\env_name\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' from 'onnxruntime.capi._pybind_state' (D:\anaconda\envs\env_name\lib\site-packages\onnxruntime\capi\_pybind_state.py)
不是onnxruntime-gpu版本过高的问题,版本是1.1.0,原因在于对应的torch为1.9,cuda版本要为10.2