Skip to content

Commit 2abe4ba

Browse files
committed
feat: support finding libraries on aarch64
Signed-off-by: Naren Dasan <[email protected]>
1 parent 7680456 commit 2abe4ba

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

py/torch_tensorrt/__init__.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,10 @@ def _find_lib(name, paths):
6060
elif sys.platform.startswith("linux"):
6161
LINUX_PATHS = [
6262
"/usr/local/cuda/lib64",
63-
] + os.environ["LD_LIBRARY_PATH"].split(os.path.pathsep)
63+
]
64+
65+
if "LD_LIBRARY_PATH" in os.environ:
66+
LINUX_PATHS += os.environ["LD_LIBRARY_PATH"].split(os.path.pathsep)
6467

6568
if platform.uname().processor == "x86_64":
6669
LINUX_PATHS += [

0 commit comments

Comments
 (0)