reduce images

ameli · Dec 20, 2023 · cb4a6af · cb4a6af
1 parent 9e50747
commit cb4a6af
Show file tree

Hide file tree

Showing 6 changed files with 37 additions and 51 deletions.
diff --git a/README.rst b/README.rst
@@ -6,6 +6,17 @@ Dockerfiles containing an installation of **NVIDIA CUDA toolkit**, built on top
 * For X86_64 architecture: `quay.io/pypa/manylinux2014_x86_64 <https://github.com/pypa/manylinux>`__
 * For AARCH64 architecture: `quay.io/pypa/manylinux2014_aarch64 <https://github.com/pypa/manylinux>`__
 
+Packages
+========
+
+The Docker images are designed specifically for building Python wheels. To maintain a minimal Docker image size, only the essential compilers and libraries from CUDA Toolkit are included. These include:
+
+* CUDA compiler: ``cuda-crt``, ``cuda-cuobjdump``, ``cuda-cuxxfilt``, ``cuda-nvcc``, ``cuda-nvprune``, ``cuda-nvvm``, ``cuda-cudart``, ``cuda-nvrtc``, ``cuda-opencl``,
+* CUDA libraries: ``libcublas``, ``libcufft``, ``libcufile``, ``libcurand``, ``libcusolver``, ``libcusparse``, ``libnpp``, ``libnvjitlink``, ``libnvjpeg``
+* CUDA development libraries: ``cuda-cccl``, ``cuda-cudart-devel``, ``cuda-driver-devel``, ``cuda-nvrtc-devel``, ``cuda-opencl-devel``, ``cuda-profiler-api``, ``libcublas-devel``, ``libcufft-devel``, ``libcufile-devel``, ``libcurand-devel``, ``libcusolver-devel``, ``libcusparse-devel``, ``libnpp-devel``, ``libnvjitlink-devel``, ``libnvjpeg-devel``
+
+If you need additional packages from CUDA toolkit to be included in the images, please feel free to create a `GitHub issue <https://github.com/ameli/manylinux-cuda/issues>`__.
+
 Download Images
 ===============
 

diff --git a/docker/manylinux2014_aarch64_cuda_12.3/Dockerfile b/docker/manylinux2014_aarch64_cuda_12.3/Dockerfile
@@ -24,19 +24,15 @@ LABEL Description="manylinux2014_aarch64 with cuda 12.3"
 # Install cuda
 # ------------
 
-# RUN yum install -y yum-utils
-# RUN yum install -y wget
-#
-# RUN wget https://developer.download.nvidia.com/compute/cuda/12.3.1/local_installers/cuda-repo-rhel9-12-3-local-12.3.1_545.23.08-1.aarch64.rpm
-# RUN rpm -i cuda-repo-rhel9-12-3-local-12.3.1_545.23.08-1.aarch64.rpm
-# RUN yum clean all
-# RUN yum -y install cuda-toolkit-12-3
-# RUN rm cuda-repo-rhel9-12-3-local-12.3.1_545.23.08-1.aarch64.rpm
+VER="12-3"
+ARCH="aarch64"
 
 RUN yum install -y yum-utils
-RUN yum config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel9/sbsa/cuda-rhel9.repo
+RUN yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel9/sbsa/cuda-rhel9.repo
 RUN yum clean all
-RUN install cuda-toolkit-12-3.aarch64
+RUN yum -y install cuda-compiler-${VER}.${ARCH} \
+                   cuda-libraries-${VER}.${ARCH} \
+                   cuda-libraries-devel-${VER}.${ARCH}
 
 # -------------------------
 # Set environment variables

diff --git a/docker/manylinux2014_x86_64_cuda_10.2/Dockerfile b/docker/manylinux2014_x86_64_cuda_10.2/Dockerfile
@@ -24,20 +24,15 @@ LABEL Description="manylinux2014_x86_64 with cuda 10.2"
 # Install cuda
 # ------------
 
-# RUN yum install -y yum-utils
-# RUN yum install -y wget
-#
-# RUN wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda-repo-rhel7-10-2-local-10.2.89-440.33.01-1.0-1.x86_64.rpm
-# RUN rpm -i cuda-repo-rhel7-10-2-local-10.2.89-440.33.01-1.0-1.x86_64.rpm
-# RUN yum clean all
-# RUN yum -y install nvidia-driver-latest-dkms cuda
-# RUN yum -y install cuda-drivers
-# RUN rm cuda-repo-rhel7-10-2-local-10.2.89-440.33.01-1.0-1.x86_64.rpm
+VER="10-2"
+ARCH="x86_64"
 
 RUN yum install -y yum-utils
 RUN yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo
 RUN yum clean all
-RUN yum -y install cuda-toolkit-10-2.x86_64
+RUN yum -y install cuda-compiler-${VER}.${ARCH} \
+                   cuda-libraries-${VER}.${ARCH} \
+                   cuda-libraries-devel-${VER}.${ARCH}
 
 # -------------------------
 # Set environment variables

diff --git a/docker/manylinux2014_x86_64_cuda_11.8/Dockerfile b/docker/manylinux2014_x86_64_cuda_11.8/Dockerfile
@@ -24,20 +24,15 @@ LABEL Description="manylinux2014_x86_64 with cuda 11.8"
 # Install cuda
 # ------------
 
-# RUN yum install -y yum-utils
-# RUN yum install -y wget
-#
-# RUN wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-rhel7-11-8-local-11.8.0_520.61.05-1.x86_64.rpm
-# RUN rpm -i cuda-repo-rhel7-11-8-local-11.8.0_520.61.05-1.x86_64.rpm
-# RUN yum clean all
-# RUN yum -y install nvidia-driver-latest-dkms cuda
-# RUN yum -y install cuda-drivers
-# RUN rm cuda-repo-rhel7-11-8-local-11.8.0_520.61.05-1.x86_64.rpm
+VER="11-8"
+ARCH="x86_64"
 
 RUN yum install -y yum-utils
 RUN yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo
 RUN yum clean all
-RUN yum -y install cuda-toolkit-11-8.x86_64
+RUN yum -y install cuda-compiler-${VER}.${ARCH} \
+                   cuda-libraries-${VER}.${ARCH} \
+                   cuda-libraries-devel-${VER}.${ARCH}
 
 # -------------------------
 # Set environment variables

diff --git a/docker/manylinux2014_x86_64_cuda_12.0/Dockerfile b/docker/manylinux2014_x86_64_cuda_12.0/Dockerfile
@@ -24,20 +24,15 @@ LABEL Description="manylinux2014_x86_64 with cuda 12.0"
 # Install cuda
 # ------------
 
-# RUN yum install -y yum-utils
-# RUN yum install -y wget
-#
-# RUN wget https://developer.download.nvidia.com/compute/cuda/12.0.1/local_installers/cuda-repo-rhel7-12-0-local-12.0.1_525.85.12-1.x86_64.rpm
-# RUN rpm -i cuda-repo-rhel7-12-0-local-12.0.1_525.85.12-1.x86_64.rpm
-# RUN yum clean all
-# RUN yum -y install nvidia-driver-latest-dkms cuda
-# RUN yum -y install cuda-drivers
-# RUN rm cuda-repo-rhel7-12-0-local-12.0.1_525.85.12-1.x86_64.rpm
+VER="12-0"
+ARCH="x86_64"
 
 RUN yum install -y yum-utils
 RUN yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo
 RUN yum clean all
-RUN yum -y install cuda-toolkit-12-0.x86_64
+RUN yum -y install cuda-compiler-${VER}.${ARCH} \
+                   cuda-libraries-${VER}.${ARCH} \
+                   cuda-libraries-devel-${VER}.${ARCH}
 
 # -------------------------
 # Set environment variables

diff --git a/docker/manylinux2014_x86_64_cuda_12.3/Dockerfile b/docker/manylinux2014_x86_64_cuda_12.3/Dockerfile
@@ -24,21 +24,15 @@ LABEL Description="manylinux2014_x86_64 with cuda 12.3"
 # Install cuda
 # ------------
 
-# RUN yum install -y yum-utils
-# RUN yum install -y wget
-#
-# RUN wget https://developer.download.nvidia.com/compute/cuda/12.3.0/local_installers/cuda-repo-rhel7-12.3-local-12.3.0_535.54.03-1.x86_64.rpm
-# RUN rpm -i cuda-repo-rhel7-12.3-local-12.3.0_535.54.03-1.x86_64.rpm
-# RUN yum clean all
-# RUN yum -y install nvidia-driver-latest-dkms cuda
-# RUN yum -y install cuda-drivers
-# RUN rm -f cuda-repo-rhel7-12.3-local-12.3.0_535.54.03-1.x86_64.rpm
-# RUN rm -rf /var/cuda-repo-rhel7-12.3-local
+VER="12-3"
+ARCH="x86_64"
 
 RUN yum install -y yum-utils
 RUN yum-config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo
 RUN yum clean all
-RUN yum -y install cuda-toolkit-12-3.x86_64
+RUN yum -y install cuda-compiler-${VER}.${ARCH} \
+                   cuda-libraries-${VER}.${ARCH} \
+                   cuda-libraries-devel-${VER}.${ARCH}
 
 # -------------------------
 # Set environment variables