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 b206884 commit de5b8d7Copy full SHA for de5b8d7
py/torch_tensorrt/_utils.py
@@ -1,6 +1,5 @@
1
-from typing import Any
2
import sys
3
-import platform
+from typing import Any
4
5
import torch
6
@@ -16,9 +15,9 @@ def sanitized_torch_version() -> Any:
16
15
def check_cross_compile_trt_win_lib() -> bool:
17
# cross compile feature is only available on linux
18
# build engine on linux and run on windows
19
- import dllist
20
-
21
if sys.platform.startswith("linux"):
+ import dllist
+
22
loaded_libs = dllist.dllist()
23
target_lib = "libnvinfer_builder_resource_win.so.*"
24
if target_lib in loaded_libs:
0 commit comments