35 lines
1.5 KiB
Docker
35 lines
1.5 KiB
Docker
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 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 && \
|
|
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-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
|
|
RUN wget -nv https://github.com/Kitware/CMake/releases/download/v3.24.2/cmake-3.24.2-linux-x86_64.sh && \
|
|
mkdir /opt/cmake && sh ./cmake-3.24.2-linux-x86_64.sh --skip-license --prefix=/opt/cmake && ln -s /opt/cmake/bin/cmake /usr/bin/cmake && \
|
|
rm cmake-3.24.2-linux-x86_64.sh
|
|
|
|
|
|
RUN echo "nameserver 8.8.8.8" >> /etc/resolv.conf
|
|
RUN echo "nameserver 9.9.9.9" >> /etc/resolv.conf
|
|
|