We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8a7d893 commit f1d58c5Copy full SHA for f1d58c5
py/torch_tensorrt/_utils.py
@@ -1,7 +1,8 @@
1
from typing import Any
2
+import sys
3
+import platform
4
5
import torch
-from torch_tensorrt._enums import Platform
6
7
8
def sanitized_torch_version() -> Any:
@@ -17,9 +18,7 @@ def check_cross_compile_trt_win_lib() -> bool:
17
18
# build engine on linux and run on windows
19
import dllist
20
- platform = Platform.current_platform()
21
- platform = str(platform).lower()
22
- if platform.startswith("linux"):
+ if sys.platform.startswith("linux"):
23
loaded_libs = dllist.dllist()
24
target_lib = "libnvinfer_builder_resource_win.so.*"
25
if target_lib in loaded_libs:
0 commit comments