add cuda installation

MBI-Div-B · Jun 18, 2023 · a39c2d9 · a39c2d9
1 parent fbe4dbc
commit a39c2d9
Showing 1 changed file with 106 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -1,5 +1,6 @@
 # Copyright (c) Jupyter Development Team.
 # Distributed under the terms of the Modified BSD License.
+# uses ubuntu:22.04 as the root container.
 ARG BASE_CONTAINER=jupyter/minimal-notebook
 FROM $BASE_CONTAINER
 
@@ -24,7 +25,9 @@ RUN apt-get update --yes && \
     # for slsdetectorlib
     cmake \
     libzmq5-dev \
-    libtiff5-dev \ 
+    libtiff5-dev \
+    # for cuda installation
+    apt-utils \
     # user requests
     imagemagick \
     # for matplotlib anim
@@ -34,6 +37,108 @@ RUN apt-get update --yes && \
 RUN /tmp/scripts/install-mbistyles-fonts.sh
 RUN /tmp/scripts/install-slsdetector-package.sh
 
+# cuda installation like https://hub.docker.com/layers/nvidia/cuda/11.7.0-cudnn8-devel-ubuntu22.04/images/sha256-cf262ed40aa17b168184fd9a8d0ac2ae932e2c6d0fe4f08e0598ab50db08a07d?context=explore
+# took instructions from its installation steps 
+
+ENV TARGETARCH=amd64
+ENV NVARCH=x86_64
+ENV NVIDIA_REQUIRE_CUDA=cuda>=11.7 brand=tesla,driver>=450,driver<451 brand=tesla,driver>=470,driver<471 brand=unknown,driver>=470,driver<471 brand=nvidia,driver>=470,driver<471 brand=nvidiartx,driver>=470,driver<471 brand=geforce,driver>=470,driver<471 brand=geforcertx,driver>=470,driver<471 brand=quadro,driver>=470,driver<471 brand=quadrortx,driver>=470,driver<471 brand=titan,driver>=470,driver<471 brand=titanrtx,driver>=470,driver<471 brand=tesla,driver>=510,driver<511 brand=unknown,driver>=510,driver<511 brand=nvidia,driver>=510,driver<511 brand=nvidiartx,driver>=510,driver<511 brand=quadro,driver>=510,driver<511 brand=quadrortx,driver>=510,driver<511 brand=titan,driver>=510,driver<511 brand=titanrtx,driver>=510,driver<511 brand=geforce,driver>=510,driver<511 brand=geforcertx,driver>=510,driver<511
+ENV NV_CUDA_CUDART_VERSION=11.7.60-1
+ENV NV_CUDA_LIB_VERSION=11.7.0-1
+ENV NV_CUDA_COMPAT_PACKAGE=cuda-compat-11-7
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    gnupg2 curl ca-certificates && \
+    curl -fsSLO https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.0-1_all.deb && \
+    dpkg -i cuda-keyring_1.0-1_all.deb && \
+    rm cuda-keyring_1.0-1_all.deb
+
+ENV CUDA_VERSION=11.7.0
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    cuda-cudart-11-7=${NV_CUDA_CUDART_VERSION} \
+    ${NV_CUDA_COMPAT_PACKAGE}
+
+RUN echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && \
+    echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf
+ENV PATH=$PATH:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
+ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/nvidia/lib:/usr/local/nvidia/lib64
+ENV NVIDIA_VISIBLE_DEVICES=all
+ENV NVIDIA_DRIVER_CAPABILITIES=compute,utility
+ENV NV_CUDA_LIB_VERSION=11.7.0-1
+ENV NV_NVTX_VERSION=11.7.50-1
+ENV NV_LIBNPP_VERSION=11.7.3.21-1
+ENV NV_LIBNPP_PACKAGE=libnpp-11-7=11.7.3.21-1
+ENV NV_LIBCUSPARSE_VERSION=11.7.3.50-1
+ENV NV_LIBCUBLAS_PACKAGE_NAME=libcublas-11-7
+ENV NV_LIBCUBLAS_VERSION=11.10.1.25-1
+ENV NV_LIBCUBLAS_PACKAGE=libcublas-11-7=11.10.1.25-1
+ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
+ENV NV_LIBNCCL_PACKAGE_VERSION=2.13.4-1
+ENV NCCL_VERSION=2.13.4-1
+ENV NV_LIBNCCL_PACKAGE=libnccl2=2.13.4-1+cuda11.7
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    cuda-libraries-11-7=${NV_CUDA_LIB_VERSION} \
+    ${NV_LIBNPP_PACKAGE} \
+    cuda-nvtx-11-7=${NV_NVTX_VERSION} \
+    libcusparse-11-7=${NV_LIBCUSPARSE_VERSION} \
+    ${NV_LIBCUBLAS_PACKAGE} \
+    ${NV_LIBNCCL_PACKAGE}
+
+RUN apt-mark hold ${NV_LIBCUBLAS_PACKAGE_NAME} \
+    ${NV_LIBNCCL_PACKAGE_NAME}
+
+ENV NV_CUDA_CUDART_DEV_VERSION=11.7.60-1
+ENV NV_NVML_DEV_VERSION=11.7.50-1
+ENV NV_LIBCUSPARSE_DEV_VERSION=11.7.3.50-1
+ENV NV_LIBNPP_DEV_VERSION=11.7.3.21-1
+ENV NV_LIBNPP_DEV_PACKAGE=libnpp-dev-11-7=11.7.3.21-1
+ENV NV_LIBCUBLAS_DEV_VERSION=11.10.1.25-1
+ENV NV_LIBCUBLAS_DEV_PACKAGE_NAME=libcublas-dev-11-7
+ENV NV_LIBCUBLAS_DEV_PACKAGE=libcublas-dev-11-7=11.10.1.25-1
+ENV NV_NSIGHT_COMPUTE_VERSION=11.7.0-1
+ENV NV_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-11-7=11.7.0-1
+ENV NV_NVPROF_VERSION=11.7.50-1
+ENV NV_NVPROF_DEV_PACKAGE=cuda-nvprof-11-7=11.7.50-1
+ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev
+ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.13.4-1
+ENV NCCL_VERSION=2.13.4-1
+ENV NV_LIBNCCL_DEV_PACKAGE=libnccl-dev=2.13.4-1+cuda11.7
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    cuda-cudart-dev-11-7=${NV_CUDA_CUDART_DEV_VERSION} \
+    cuda-command-line-tools-11-7=${NV_CUDA_LIB_VERSION} \
+    cuda-minimal-build-11-7=${NV_CUDA_LIB_VERSION} \
+    cuda-libraries-dev-11-7=${NV_CUDA_LIB_VERSION} \
+    cuda-nvml-dev-11-7=${NV_NVML_DEV_VERSION} \
+    ${NV_NVPROF_DEV_PACKAGE} \
+    ${NV_LIBNPP_DEV_PACKAGE} \
+    libcusparse-dev-11-7=${NV_LIBCUSPARSE_DEV_VERSION} \
+    ${NV_LIBCUBLAS_DEV_PACKAGE} \
+    ${NV_LIBNCCL_DEV_PACKAGE} \
+    ${NV_NSIGHT_COMPUTE_DEV_PACKAGE}
+RUN apt-mark hold ${NV_LIBCUBLAS_DEV_PACKAGE_NAME} \
+    ${NV_LIBNCCL_DEV_PACKAGE_NAME}
+ENV LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64/stubs
+ENV NV_CUDNN_VERSION=8.5.0.96
+ENV NV_CUDNN_PACKAGE_NAME=libcudnn8
+ENV NV_CUDNN_PACKAGE=libcudnn8=8.5.0.96-1+cuda11.7
+ENV NV_CUDNN_PACKAGE_DEV=libcudnn8-dev=8.5.0.96-1+cuda11.7
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    ${NV_CUDNN_PACKAGE} \
+    ${NV_CUDNN_PACKAGE_DEV} && \
+    apt-mark hold ${NV_CUDNN_PACKAGE_NAME}
+
+# mumax installation to test if it works
+# since the cuda version is 11.7 maybe we will need to build it from sources
+RUN mkdir /opt/mumax3 && cd /opt/mumax3 && \
+    wget https://mumax.ugent.be/mumax3-binaries/mumax3.10_linux_cuda11.0.tar.gz && \
+    tar -xzf mumax3.10_linux_cuda11.0.tar.gz
+    # && \
+    #rm mumax3-2020.10.01-Linux.tar.gz && \
+    #ln -s /opt/mumax3/mumax3 /usr/local/bin/mumax3
+
+
 USER ${NB_UID}
 
 #Install Python 3 packages