From f05b75b138dd6772ab801786984cdec4c9b23dc5 Mon Sep 17 00:00:00 2001 From: HolyWu Date: Mon, 16 Jun 2025 09:53:40 +0800 Subject: [PATCH] Import dllist only on linux --- py/torch_tensorrt/_utils.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/py/torch_tensorrt/_utils.py b/py/torch_tensorrt/_utils.py index 9875ae7bbc..419de31c7e 100644 --- a/py/torch_tensorrt/_utils.py +++ b/py/torch_tensorrt/_utils.py @@ -1,6 +1,5 @@ -from typing import Any import sys -import platform +from typing import Any import torch @@ -16,9 +15,9 @@ def sanitized_torch_version() -> Any: def check_cross_compile_trt_win_lib() -> bool: # cross compile feature is only available on linux # build engine on linux and run on windows - import dllist - if sys.platform.startswith("linux"): + import dllist + loaded_libs = dllist.dllist() target_lib = "libnvinfer_builder_resource_win.so.*" if target_lib in loaded_libs: