Merge pull request #149 from iot-salzburg/v1.7_cuda-12.2_ubuntu-22.04

set TF version to 2.15 for CUDA 12.2
iot-salzburg · Jun 18, 2024 · c2cc46d · c2cc46d
2 parents 43b993f + f2cd1df
commit c2cc46d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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}"