fix tensorrt by ignoring it

ucsd-ets · Dec 29, 2024 · 2ca4743 · 2ca4743
1 parent 289470f
commit 2ca4743
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile
@@ -8,9 +8,9 @@ USER root
 # https://www.tensorflow.org/install/source#linux
 
 # Python/Mamba deps
-## tf 2.13 does not work with torch 2.2.1. Both require conflicting versions of typing-extensions
+## tf 2.18+ does not work with cu124 pytorch. tf 2.18 needs cudnn 9.3+, torch installs cudnn 9.1
 ARG CUDA_VERSION=12.4 \
-  TENSORFLOW_VERSION=2.18.0 KERAS_VERSION=3.7.0 TENSORRT_VERSION=10.7.0 TORCH_VERSION=2.5.1
+  TENSORFLOW_VERSION=2.17.0 KERAS_VERSION=3.7.0 TENSORRT_VERSION=10.7.0 TORCH_VERSION=2.5.1
 
 # apt deps
 RUN apt-get update && \

diff --git a/images/scipy-ml-notebook/tensorrt_env_vars.sh b/images/scipy-ml-notebook/tensorrt_env_vars.sh
@@ -4,6 +4,6 @@
 #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:$LD_LIBRARY_PATH"
-export LD_LIBRARY_PATH="/opt/conda/lib/python3.11/site-packages/tensorrt_bindings:$LD_LIBRARY_PATH"
-export LD_LIBRARY_PATH="/opt/conda/lib/python3.11/site-packages/tensorrt_libs:$LD_LIBRARY_PATH"
+export LD_LIBRARY_PATH="/opt/conda/lib/python3.12/site-packages/tensorrt:$LD_LIBRARY_PATH"
+export LD_LIBRARY_PATH="/opt/conda/lib/python3.12/site-packages/tensorrt_bindings:$LD_LIBRARY_PATH"
+export LD_LIBRARY_PATH="/opt/conda/lib/python3.12/site-packages/tensorrt_libs:$LD_LIBRARY_PATH"