Fixed tensorrt plugin not found in Windows (#672)

open-mmlab · Jun 29, 2022 · 0f0f676 · 0f0f676
1 parent 30d62e9
commit 0f0f676
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mmdeploy/backend/tensorrt/init_plugins.py b/mmdeploy/backend/tensorrt/init_plugins.py
@@ -13,7 +13,9 @@ def get_ops_path() -> str:
     """
     candidates = [
         '../../lib/libmmdeploy_tensorrt_ops.so',
-        '../../lib/mmdeploy_tensorrt_ops.dll'
+        '../../lib/mmdeploy_tensorrt_ops.dll',
+        '../../../build/lib/libmmdeploy_tensorrt_ops.so',
+        '../../../build/bin/*/mmdeploy_tensorrt_ops.dll'
     ]
     return get_file_path(os.path.dirname(__file__), candidates)