From 4e4265c5c9d1ff185af9a3e0888dbe9bc09d9d32 Mon Sep 17 00:00:00 2001 From: brian Date: Tue, 15 Aug 2023 09:23:18 +0200 Subject: [PATCH] Update Docker image with cudnn 8.9.4.25 --- .docker/Dockerfile | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/.docker/Dockerfile b/.docker/Dockerfile index 6759d8ae8..628583766 100644 --- a/.docker/Dockerfile +++ b/.docker/Dockerfile @@ -4,19 +4,21 @@ ENV OS=ubuntu2004 ENV cudnn_version=8.9.4.25 ENV cuda_version=cuda12.2 +RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y wget + RUN wget https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/cuda-${OS}.pin RUN mv cuda-${OS}.pin /etc/apt/preferences.d/cuda-repository-pin-600 RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/7fa2af80.pub RUN add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/${OS}/x86_64/ /" -RUN apt-get update + RUN apt-get update && \ - DEBIAN_FRONTEND=noninteractive apt-get install -y openjdk-11-jdk wget \ - build-essential checkinstall zlib1g-dev libssl-dev git libpthread-stubs0-dev + DEBIAN_FRONTEND=noninteractive apt-get install -y openjdk-11-jdk \ + build-essential checkinstall zlib1g-dev libssl-dev git libpthread-stubs0-dev \ + libcudnn8=${cudnn_version}-1+${cuda_version} libcudnn8-dev=${cudnn_version}-1+${cuda_version} -RUN apt-get install libcudnn8=${cudnn_version}-1+${cuda_version} libcudnn8-dev=${cudnn_version}-1+${cuda_version} #RUN apt-get install libcudnn8-samples=${cudnn_version}-1+${cuda_version} #Build cmake version from source \ #RUN wget https://github.com/Kitware/CMake/releases/download/v3.24.2/cmake-3.24.2.tar.gz && \