LD_LIBRARY_PATH for linux

2023-05-28 00:37:29 -04:00 · 2023-05-28 00:37:29 -04:00 · 57b853cc73
parent 151b77db70
commit 57b853cc73
1 changed files with 5 additions and 0 deletions
--- a/trt_paths.py
+++ b/trt_paths.py
@ -30,6 +30,11 @@ def set_paths():
    assert trt_path is not None, "Was not able to find TensorRT directory. Looked in: " + ", ".join(looked_in)

    trt_lib_path = os.path.join(trt_path, "lib")
+    if "LD_LIBRARY_PATH" in os.environ:
+        os.environ["LD_LIBRARY_PATH"] = f'{trt_lib_path}:{os.environ["LD_LIBRARY_PATH"]}'
+    elif os.name == 'posix':
+        os.environ["LD_LIBRARY_PATH"] = trt_lib_path
+
    if trt_lib_path not in os.environ['PATH']:
        os.environ['PATH'] = os.environ['PATH'] + os.pathsep + trt_lib_path