Attempt to fix tensorrt as well, locations changed with 8.6+

ucsd-ets · Mar 3, 2024 · 04b87c5 · 04b87c5
1 parent f3a3fa9
commit 04b87c5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/images/scipy-ml-notebook/tensorrt_env_vars.sh b/images/scipy-ml-notebook/tensorrt_env_vars.sh
@@ -1,4 +1,6 @@
 #!/bin/bash
 
 # put tensorrt lib into LD_LIBRARY_PATH
-export LD_LIBRARY_PATH="/opt/conda/lib/python3.11/site-packages/tensorrt:$LD_LIBRARY_PATH"
+#export LD_LIBRARY_PATH="/opt/conda/lib/python3.11/site-packages/tensorrt:$LD_LIBRARY_PATH"
+# tensorrt 8.6+?
+export LD_LIBRARY_PATH="/opt/conda/lib/python3.11/site-packages/tensorrt_libs:$LD_LIBRARY_PATH"