From 9f813882370d22e7e239d1cd58e85a28ff0fa178 Mon Sep 17 00:00:00 2001
From: Henning Becker <43133967+beckerhe@users.noreply.github.com>
Date: Fri, 17 Nov 2023 20:44:11 +0100
Subject: [PATCH] Update CUDA to 12.3 in tf_runtime Docker images (#165)

---
 tensorflow_runtime_dockerfiles/gpu.Dockerfile |  6 ++--
 .../gpu.packages.txt                          | 30 +++++++++----------
 2 files changed, 18 insertions(+), 18 deletions(-)

diff --git a/tensorflow_runtime_dockerfiles/gpu.Dockerfile b/tensorflow_runtime_dockerfiles/gpu.Dockerfile
index 578d5f0..906ab33 100644
--- a/tensorflow_runtime_dockerfiles/gpu.Dockerfile
+++ b/tensorflow_runtime_dockerfiles/gpu.Dockerfile
@@ -13,7 +13,7 @@
 # limitations under the License.
 # ============================================================================
 
-FROM nvidia/cuda:11.8.0-base-ubuntu22.04 as base
+FROM nvidia/cuda:12.3.0-base-ubuntu22.04 as base
 ENV DEBIAN_FRONTEND=noninteractive
 ENV LANG C.UTF-8
 
@@ -52,5 +52,5 @@ CMD ["bash", "-c", "source /etc/bash.bashrc && jupyter notebook --notebook-dir=/
 FROM base as test
 
 ENV LD_LIBRARY_PATH /usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH
-COPY test.import_cpu.sh /test.import_cpu.sh
-RUN /test.import_cpu.sh
+COPY test.import_gpu.sh /test.import_gpu.sh
+RUN /test.import_gpu.sh
diff --git a/tensorflow_runtime_dockerfiles/gpu.packages.txt b/tensorflow_runtime_dockerfiles/gpu.packages.txt
index 59d10e3..0101050 100644
--- a/tensorflow_runtime_dockerfiles/gpu.packages.txt
+++ b/tensorflow_runtime_dockerfiles/gpu.packages.txt
@@ -1,20 +1,20 @@
 # All required CUDA packages
-cuda-command-line-tools-11-8
-cuda-cudart-dev-11-8
-cuda-nvcc-11-8
-cuda-cupti-11-8
-cuda-nvprune-11-8
-cuda-libraries-11-8
-cuda-nvrtc-11-8
-libcufft-11-8
-libcurand-11-8
-libcusolver-11-8
-libcusparse-11-8
-libcublas-11-8
+cuda-command-line-tools-12-3
+cuda-cudart-dev-12-3
+cuda-nvcc-12-3
+cuda-cupti-12-3
+cuda-nvprune-12-3
+cuda-libraries-12-3
+cuda-nvrtc-12-3
+libcufft-12-3
+libcurand-12-3
+libcusolver-12-3
+libcusparse-12-3
+libcublas-12-3
 # CuDNN: https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#ubuntu-network-installation
-libcudnn8=8.6.0.163-1+cuda11.8
-libnvinfer-plugin8=8.6.1.6-1+cuda11.8
-libnvinfer8=8.6.1.6-1+cuda11.8
+libcudnn8=8.9.6.50-1+cuda12.2
+libnvinfer-plugin8=8.6.1.6-1+cuda12.0
+libnvinfer8=8.6.1.6-1+cuda12.0
 
 # Other packages
 build-essential
-- 
GitLab