From 796e3a6be0f524de12650b2696e09a5759851894 Mon Sep 17 00:00:00 2001 From: brian Date: Tue, 15 Aug 2023 09:19:32 +0200 Subject: [PATCH] Update Docker image with cudnn 8.9.4.25 Signed-off-by: brian --- .docker/Dockerfile | 17 ++++++++++++++++- 1 file changed, 16 insertions(+), 1 deletion(-) diff --git a/.docker/Dockerfile b/.docker/Dockerfile index f17ee13ad..2304329d3 100644 --- a/.docker/Dockerfile +++ b/.docker/Dockerfile @@ -1,8 +1,23 @@ FROM nvidia/cuda:12.2.0-devel-ubuntu20.04 +ENV OS=ubuntu2004 +ENV cudnn_version=8.9.4.25 +ENV cuda_version=cuda12.2 + +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 -#Build cmake version from source \ + + +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 && \ # tar -xvf cmake-3.24.2.tar.gz && cd cmake-3.24.2 && \ # ./bootstrap && make && make install