chore: updates

Author: peri044

packaging/pre_build_script.sh

@@ -3,7 +3,7 @@
 # Install dependencies
 python3 -m pip install pyyaml
 yum install -y ninja-build gettext
-TRT_VERSION=$(python3 -c "import versions; versions.tensorrt_version()")
+TRT_VERSION=10.0.0.6 #$(python3 -c "import versions; versions.tensorrt_version()")
 wget -P /opt/torch-tensorrt-builds/ https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.0.0/TensorRT-10.0.0.6.Linux.x86_64-gnu.cuda-12.4.tar.gz
 tar -xvzf /opt/torch-tensorrt-builds/TensorRT-10.0.0.6.Linux.x86_64-gnu.cuda-12.4.tar.gz -C /opt/torch-tensorrt-builds/
 export LD_LIBRARY_PATH=/opt/torch-tensorrt-builds/TensorRT-10.0.0.6/lib:$LD_LIBRARY_PATH

py/torch_tensorrt/__init__.py

@@ -60,7 +60,7 @@ def _find_lib(name: str, paths: List[str]) -> str:
 
     elif sys.platform.startswith("linux"):
         LINUX_PATHS = ["/usr/local/cuda-12.1/lib64", "/usr/lib", "/usr/lib64"]
-
+        print("==== BOOL CHECK", "LD_LIBRARY_PATH" in os.environ)
         if "LD_LIBRARY_PATH" in os.environ:
             LINUX_PATHS += os.environ["LD_LIBRARY_PATH"].split(os.path.pathsep)
 
@@ -92,9 +92,8 @@ def _find_lib(name: str, paths: List[str]) -> str:
 from torch_tensorrt.runtime import *  # noqa: F403
 
 if version.parse(sanitized_torch_version()) >= version.parse("2.1.dev"):
-    from torch_tensorrt.dynamo import backend  # noqa: F401
-
     from torch_tensorrt import dynamo  # noqa: F401
+    from torch_tensorrt.dynamo import backend  # noqa: F401
 
 
 def _register_with_torch() -> None: