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