diff --git a/mmdeploy/backend/onnxruntime/init_plugins.py b/mmdeploy/backend/onnxruntime/init_plugins.py index 72d1115d98..fd0d850fe5 100644 --- a/mmdeploy/backend/onnxruntime/init_plugins.py +++ b/mmdeploy/backend/onnxruntime/init_plugins.py @@ -24,7 +24,7 @@ def get_lib_path() -> str: str: The library path to onnxruntime. """ candidates = [ - '../../lib/libonnxruntime_ops.so*', + '../../lib/libonnxruntime.so*', '../../lib/onnxruntime.dll', ] return get_file_path(os.path.dirname(__file__), candidates) diff --git a/mmdeploy/backend/onnxruntime/wrapper.py b/mmdeploy/backend/onnxruntime/wrapper.py index f866283aae..bd83d7f32d 100644 --- a/mmdeploy/backend/onnxruntime/wrapper.py +++ b/mmdeploy/backend/onnxruntime/wrapper.py @@ -48,7 +48,7 @@ def __init__(self, # load ort lib before custom ops lib lib_path = get_lib_path() if osp.exists(lib_path): - ctypes.DLL(lib_path) + ctypes.CDLL(lib_path) session_options.register_custom_ops_library(ort_custom_op_path) logger.info('Successfully loaded onnxruntime custom ops from '