From 57b853cc732322d6a31a2d78bd04b51c14d68aea Mon Sep 17 00:00:00 2001 From: Mackerel Date: Sun, 28 May 2023 00:37:29 -0400 Subject: [PATCH] LD_LIBRARY_PATH for linux --- trt_paths.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/trt_paths.py b/trt_paths.py index f76e840..47c407d 100644 --- 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