diff --git a/images/scipy-ml-notebook/tensorrt_env_vars.sh b/images/scipy-ml-notebook/tensorrt_env_vars.sh index d255f57b..c84cc65b 100644 --- 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"