From 2ca474347794375fdfe28c09bb151e1f7b7cc4f0 Mon Sep 17 00:00:00 2001 From: dafeliton Date: Sun, 29 Dec 2024 01:52:21 -0800 Subject: [PATCH] fix tensorrt by ignoring it --- images/scipy-ml-notebook/Dockerfile | 4 ++-- images/scipy-ml-notebook/tensorrt_env_vars.sh | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/images/scipy-ml-notebook/Dockerfile b/images/scipy-ml-notebook/Dockerfile index 487f367e..f3055c36 100644 --- 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 index a1df0eb9..2e3f579a 100644 --- 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" \ No newline at end of file +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" \ No newline at end of file