From f2cd1dfc49dd940177264a24288d59af05b6a3ee Mon Sep 17 00:00:00 2001 From: Christoph Schranz Date: Tue, 18 Jun 2024 15:47:10 +0200 Subject: [PATCH] set TF version to 2.15 for CUDA 12.2 --- .build/Dockerfile | 2 +- README.md | 2 +- custom/gpulibs.Dockerfile | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.build/Dockerfile b/.build/Dockerfile index 40e78dd..7c0863e 100755 --- a/.build/Dockerfile +++ b/.build/Dockerfile @@ -399,7 +399,7 @@ RUN mamba install --quiet --yes \ # using device_lib.list_local_devices() the cudNN version is shown, adapt version to tested compat USER ${NB_UID} RUN pip install --upgrade pip && \ - pip install --no-cache-dir tensorflow==2.16.1 keras==3.1.1 && \ + pip install --no-cache-dir tensorflow==2.15.1 keras==2.15.0 && \ fix-permissions "${CONDA_DIR}" && \ fix-permissions "/home/${NB_USER}" diff --git a/README.md b/README.md index a7949af..83d3948 100644 --- a/README.md +++ b/README.md @@ -50,7 +50,7 @@ for creating and maintaining a robust Python, R, and Julia toolstack for Data Sc ```bash Tue Jun 18 12:39:31 2024 +---------------------------------------------------------------------------------------+ - | NVIDIA-SMI 545.23.05 Driver Version: 545.84 CUDA Version: 12.3 | + | NVIDIA-SMI 545.23.05 Driver Version: 545.84 CUDA Version: 12.2 | |-----------------------------------------+----------------------+----------------------+ | GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. | diff --git a/custom/gpulibs.Dockerfile b/custom/gpulibs.Dockerfile index 7b98b30..26ac028 100644 --- a/custom/gpulibs.Dockerfile +++ b/custom/gpulibs.Dockerfile @@ -14,7 +14,7 @@ RUN mamba install --quiet --yes \ # using device_lib.list_local_devices() the cudNN version is shown, adapt version to tested compat USER ${NB_UID} RUN pip install --upgrade pip && \ - pip install --no-cache-dir tensorflow==2.16.1 keras==3.1.1 && \ + pip install --no-cache-dir tensorflow==2.15.1 keras==2.15.0 && \ fix-permissions "${CONDA_DIR}" && \ fix-permissions "/home/${NB_USER}"