Skip to content

Commit 041f6a3

Browse files
committed
chore: updates
1 parent 96a8bf6 commit 041f6a3

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

packaging/post_build_script.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,3 @@
11
export LD_LIBRARY_PATH=/opt/torch-tensorrt-builds/TensorRT-10.0.0.6/lib:$LD_LIBRARY_PATH
2+
echo $LD_LIBRARY_PATH
23
python -m pip install /opt/torch-tensorrt-builds/TensorRT-10.0.0.6/python/tensorrt-10.0.0b6-cp38-none-linux_x86_64.whl

py/torch_tensorrt/__init__.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,12 @@ def _find_lib(name: str, paths: List[str]) -> str:
6060

6161
elif sys.platform.startswith("linux"):
6262
LINUX_PATHS = ["/usr/local/cuda-12.1/lib64", "/usr/lib", "/usr/lib64"]
63-
print("==== BOOL CHECK", "LD_LIBRARY_PATH" in os.environ)
63+
print(
64+
"==== BOOL CHECK",
65+
"LD_LIBRARY_PATH" in os.environ,
66+
" path: ",
67+
os.environ["LD_LIBRARY_PATH"],
68+
)
6469
if "LD_LIBRARY_PATH" in os.environ:
6570
LINUX_PATHS += os.environ["LD_LIBRARY_PATH"].split(os.path.pathsep)
6671

0 commit comments

Comments
 (0)