From 1ff151d89a8e4d8b62fb6f2578c3008bb9a3d8f1 Mon Sep 17 00:00:00 2001 From: brian Date: Tue, 29 Aug 2023 18:47:54 +0200 Subject: [PATCH] Reorganising build.gradle for CUDA 12 Signed-off-by: brian --- .docker/Dockerfile | 24 ++++++++++++------------ 1 file changed, 12 insertions(+), 12 deletions(-) diff --git a/.docker/Dockerfile b/.docker/Dockerfile index 86359f1d9..7169918cc 100644 --- a/.docker/Dockerfile +++ b/.docker/Dockerfile @@ -1,23 +1,23 @@ FROM nvidia/cuda:12.2.0-devel-ubuntu20.04 -ARG OS=ubuntu2004 -ARG cudnn_version=8.9.4.25 -ARG cuda_version=cuda12.2 -ARG cmake_version=3.27.4 +ENV OS=ubuntu2004 +ENV cudnn_version=8.9.4.25 +ENV cuda_version=cuda12.2 +ENV CMAKE_VER=3.27.4 RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y wget software-properties-common -RUN wget https://developer.download.nvidia.com/compute/cuda/repos/$OS/x86_64/cuda-$OS.pin +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 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 && apt-get upgrade -y && \ 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 + libcudnn8=${cudnn_version}-1+${cuda_version} libcudnn8-dev=${cudnn_version}-1+${cuda_version} #RUN apt-get install libcudnn8-samples=${cudnn_version}-1+${cuda_version} @@ -25,9 +25,9 @@ RUN apt-get update && apt-get upgrade -y && \ #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 -RUN wget -nv https://github.com/Kitware/CMake/releases/download/v$cmake_version/cmake-$cmake_version-linux-x86_64.sh && \ - mkdir -p /opt/cmake && sh ./cmake-$cmake_version-linux-x86_64.sh --skip-license --prefix=/opt/cmake && ln -s /opt/cmake/bin/cmake /usr/bin/cmake && \ - rm cmake-$cmake_version-linux-x86_64.sh +RUN wget -nv https://github.com/Kitware/CMake/releases/download/v${CMAKE_VER}/cmake-${CMAKE_VER}-linux-x86_64.sh && \ + mkdir -p /opt/cmake && sh ./cmake-${CMAKE_VER}-linux-x86_64.sh --skip-license --prefix=/opt/cmake && ln -s /opt/cmake/bin/cmake /usr/bin/cmake && \ + rm cmake-${CMAKE_VER}-linux-x86_64.sh RUN ln -s /usr/bin/make /usr/bin/gmake