Skip to content

Commit 1054ffc

Browse files
jiridanekjiridanek
committed
Issue #385: decrease Python 3.12 CUDA images size by installing less
1 parent 35f924c commit 1054ffc

File tree

5 files changed

+15
-145
lines changed

5 files changed

+15
-145
lines changed

‎jupyter/minimal/ubi9-python-3.12/Dockerfile.cuda

Lines changed: 3 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -98,49 +98,23 @@ RUN yum install -y \
9898
&& yum clean all \
9999
&& rm -rf /var/cache/yum/*
100100

101-
# Install CUDA devel from:
102-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/Dockerfile
103-
ENV NV_CUDA_CUDART_DEV_VERSION=12.6.77-1
104-
ENV NV_NVML_DEV_VERSION=12.6.77-1
105-
ENV NV_LIBCUBLAS_DEV_VERSION=12.6.4.1-1
106-
ENV NV_LIBNPP_DEV_VERSION=12.3.1.54-1
107-
ENV NV_LIBNPP_DEV_PACKAGE=libnpp-devel-12-6-${NV_LIBNPP_DEV_VERSION}
108-
ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-devel
109-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.23.4-1
110-
ENV NCCL_VERSION=2.23.4
111-
ENV NV_LIBNCCL_DEV_PACKAGE=${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.6
112-
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.6.3-1
113-
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-6-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
101+
# Install devel tools
114102

115103
RUN yum install -y \
116104
make \
117105
findutils \
118-
cuda-command-line-tools-12-6-${NV_CUDA_LIB_VERSION} \
119-
cuda-libraries-devel-12-6-${NV_CUDA_LIB_VERSION} \
120-
cuda-minimal-build-12-6-${NV_CUDA_LIB_VERSION} \
121-
cuda-cudart-devel-12-6-${NV_CUDA_CUDART_DEV_VERSION} \
122-
${NV_NVPROF_DEV_PACKAGE} \
123-
cuda-nvml-devel-12-6-${NV_NVML_DEV_VERSION} \
124-
libcublas-devel-12-6-${NV_LIBCUBLAS_DEV_VERSION} \
125-
${NV_LIBNPP_DEV_PACKAGE} \
126-
${NV_LIBNCCL_DEV_PACKAGE} \
127-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
128106
&& yum clean all \
129107
&& rm -rf /var/cache/yum/*
130108

131-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
132-
133-
# Install CUDA devel cudnn9 from:
134-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
109+
# Install CUDA cudnn9 from:
110+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
135111
ENV NV_CUDNN_VERSION=9.5.1.17-1
136112
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
137-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
138113

139114
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
140115

141116
RUN yum install -y \
142117
${NV_CUDNN_PACKAGE} \
143-
${NV_CUDNN_PACKAGE_DEV} \
144118
&& yum clean all \
145119
&& rm -rf /var/cache/yum/*
146120

‎jupyter/pytorch/ubi9-python-3.12/Dockerfile.cuda

Lines changed: 3 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -111,49 +111,23 @@ RUN yum install -y \
111111
&& yum clean all \
112112
&& rm -rf /var/cache/yum/*
113113

114-
# Install CUDA devel from:
115-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/Dockerfile
116-
ENV NV_CUDA_CUDART_DEV_VERSION=12.6.77-1
117-
ENV NV_NVML_DEV_VERSION=12.6.77-1
118-
ENV NV_LIBCUBLAS_DEV_VERSION=12.6.4.1-1
119-
ENV NV_LIBNPP_DEV_VERSION=12.3.1.54-1
120-
ENV NV_LIBNPP_DEV_PACKAGE=libnpp-devel-12-6-${NV_LIBNPP_DEV_VERSION}
121-
ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-devel
122-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.23.4-1
123-
ENV NCCL_VERSION=2.23.4
124-
ENV NV_LIBNCCL_DEV_PACKAGE=${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.6
125-
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.6.3-1
126-
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-6-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
114+
# Install devel tools
127115

128116
RUN yum install -y \
129117
make \
130118
findutils \
131-
cuda-command-line-tools-12-6-${NV_CUDA_LIB_VERSION} \
132-
cuda-libraries-devel-12-6-${NV_CUDA_LIB_VERSION} \
133-
cuda-minimal-build-12-6-${NV_CUDA_LIB_VERSION} \
134-
cuda-cudart-devel-12-6-${NV_CUDA_CUDART_DEV_VERSION} \
135-
${NV_NVPROF_DEV_PACKAGE} \
136-
cuda-nvml-devel-12-6-${NV_NVML_DEV_VERSION} \
137-
libcublas-devel-12-6-${NV_LIBCUBLAS_DEV_VERSION} \
138-
${NV_LIBNPP_DEV_PACKAGE} \
139-
${NV_LIBNCCL_DEV_PACKAGE} \
140-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
141119
&& yum clean all \
142120
&& rm -rf /var/cache/yum/*
143121

144-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
145-
146-
# Install CUDA devel cudnn9 from:
147-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
122+
# Install CUDA cudnn9 from:
123+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
148124
ENV NV_CUDNN_VERSION=9.5.1.17-1
149125
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
150-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
151126

152127
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
153128

154129
RUN yum install -y \
155130
${NV_CUDNN_PACKAGE} \
156-
${NV_CUDNN_PACKAGE_DEV} \
157131
&& yum clean all \
158132
&& rm -rf /var/cache/yum/*
159133

‎jupyter/tensorflow/ubi9-python-3.12/Dockerfile.cuda

Lines changed: 3 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -111,49 +111,23 @@ RUN yum install -y \
111111
&& yum clean all \
112112
&& rm -rf /var/cache/yum/*
113113

114-
# Install CUDA devel from:
115-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/Dockerfile
116-
ENV NV_CUDA_CUDART_DEV_VERSION=12.6.77-1
117-
ENV NV_NVML_DEV_VERSION=12.6.77-1
118-
ENV NV_LIBCUBLAS_DEV_VERSION=12.6.4.1-1
119-
ENV NV_LIBNPP_DEV_VERSION=12.3.1.54-1
120-
ENV NV_LIBNPP_DEV_PACKAGE=libnpp-devel-12-6-${NV_LIBNPP_DEV_VERSION}
121-
ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-devel
122-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.23.4-1
123-
ENV NCCL_VERSION=2.23.4
124-
ENV NV_LIBNCCL_DEV_PACKAGE=${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.6
125-
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.6.3-1
126-
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-6-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
114+
# Install devel tools
127115

128116
RUN yum install -y \
129117
make \
130118
findutils \
131-
cuda-command-line-tools-12-6-${NV_CUDA_LIB_VERSION} \
132-
cuda-libraries-devel-12-6-${NV_CUDA_LIB_VERSION} \
133-
cuda-minimal-build-12-6-${NV_CUDA_LIB_VERSION} \
134-
cuda-cudart-devel-12-6-${NV_CUDA_CUDART_DEV_VERSION} \
135-
${NV_NVPROF_DEV_PACKAGE} \
136-
cuda-nvml-devel-12-6-${NV_NVML_DEV_VERSION} \
137-
libcublas-devel-12-6-${NV_LIBCUBLAS_DEV_VERSION} \
138-
${NV_LIBNPP_DEV_PACKAGE} \
139-
${NV_LIBNCCL_DEV_PACKAGE} \
140-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
141119
&& yum clean all \
142120
&& rm -rf /var/cache/yum/*
143121

144-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
145-
146-
# Install CUDA devel cudnn9 from:
147-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
122+
# Install CUDA cudnn9 from:
123+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
148124
ENV NV_CUDNN_VERSION=9.5.1.17-1
149125
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
150-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
151126

152127
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
153128

154129
RUN yum install -y \
155130
${NV_CUDNN_PACKAGE} \
156-
${NV_CUDNN_PACKAGE_DEV} \
157131
&& yum clean all \
158132
&& rm -rf /var/cache/yum/*
159133

‎runtimes/pytorch/ubi9-python-3.12/Dockerfile.cuda

Lines changed: 3 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -98,49 +98,23 @@ RUN yum install -y \
9898
&& yum clean all \
9999
&& rm -rf /var/cache/yum/*
100100

101-
# Install CUDA devel from:
102-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/Dockerfile
103-
ENV NV_CUDA_CUDART_DEV_VERSION=12.6.77-1
104-
ENV NV_NVML_DEV_VERSION=12.6.77-1
105-
ENV NV_LIBCUBLAS_DEV_VERSION=12.6.4.1-1
106-
ENV NV_LIBNPP_DEV_VERSION=12.3.1.54-1
107-
ENV NV_LIBNPP_DEV_PACKAGE=libnpp-devel-12-6-${NV_LIBNPP_DEV_VERSION}
108-
ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-devel
109-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.23.4-1
110-
ENV NCCL_VERSION=2.23.4
111-
ENV NV_LIBNCCL_DEV_PACKAGE=${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.6
112-
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.6.3-1
113-
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-6-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
101+
# Install devel tools
114102

115103
RUN yum install -y \
116104
make \
117105
findutils \
118-
cuda-command-line-tools-12-6-${NV_CUDA_LIB_VERSION} \
119-
cuda-libraries-devel-12-6-${NV_CUDA_LIB_VERSION} \
120-
cuda-minimal-build-12-6-${NV_CUDA_LIB_VERSION} \
121-
cuda-cudart-devel-12-6-${NV_CUDA_CUDART_DEV_VERSION} \
122-
${NV_NVPROF_DEV_PACKAGE} \
123-
cuda-nvml-devel-12-6-${NV_NVML_DEV_VERSION} \
124-
libcublas-devel-12-6-${NV_LIBCUBLAS_DEV_VERSION} \
125-
${NV_LIBNPP_DEV_PACKAGE} \
126-
${NV_LIBNCCL_DEV_PACKAGE} \
127-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
128106
&& yum clean all \
129107
&& rm -rf /var/cache/yum/*
130108

131-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
132-
133-
# Install CUDA devel cudnn9 from:
134-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
109+
# Install CUDA cudnn9 from:
110+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
135111
ENV NV_CUDNN_VERSION=9.5.1.17-1
136112
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
137-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
138113

139114
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
140115

141116
RUN yum install -y \
142117
${NV_CUDNN_PACKAGE} \
143-
${NV_CUDNN_PACKAGE_DEV} \
144118
&& yum clean all \
145119
&& rm -rf /var/cache/yum/*
146120

‎runtimes/tensorflow/ubi9-python-3.12/Dockerfile.cuda

Lines changed: 3 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -98,49 +98,23 @@ RUN yum install -y \
9898
&& yum clean all \
9999
&& rm -rf /var/cache/yum/*
100100

101-
# Install CUDA devel from:
102-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/Dockerfile
103-
ENV NV_CUDA_CUDART_DEV_VERSION=12.6.77-1
104-
ENV NV_NVML_DEV_VERSION=12.6.77-1
105-
ENV NV_LIBCUBLAS_DEV_VERSION=12.6.4.1-1
106-
ENV NV_LIBNPP_DEV_VERSION=12.3.1.54-1
107-
ENV NV_LIBNPP_DEV_PACKAGE=libnpp-devel-12-6-${NV_LIBNPP_DEV_VERSION}
108-
ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-devel
109-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.23.4-1
110-
ENV NCCL_VERSION=2.23.4
111-
ENV NV_LIBNCCL_DEV_PACKAGE=${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.6
112-
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.6.3-1
113-
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-6-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
101+
# Install devel tools
114102

115103
RUN yum install -y \
116104
make \
117105
findutils \
118-
cuda-command-line-tools-12-6-${NV_CUDA_LIB_VERSION} \
119-
cuda-libraries-devel-12-6-${NV_CUDA_LIB_VERSION} \
120-
cuda-minimal-build-12-6-${NV_CUDA_LIB_VERSION} \
121-
cuda-cudart-devel-12-6-${NV_CUDA_CUDART_DEV_VERSION} \
122-
${NV_NVPROF_DEV_PACKAGE} \
123-
cuda-nvml-devel-12-6-${NV_NVML_DEV_VERSION} \
124-
libcublas-devel-12-6-${NV_LIBCUBLAS_DEV_VERSION} \
125-
${NV_LIBNPP_DEV_PACKAGE} \
126-
${NV_LIBNCCL_DEV_PACKAGE} \
127-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
128106
&& yum clean all \
129107
&& rm -rf /var/cache/yum/*
130108

131-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
132-
133-
# Install CUDA devel cudnn9 from:
134-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
109+
# Install CUDA cudnn9 from:
110+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
135111
ENV NV_CUDNN_VERSION=9.5.1.17-1
136112
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
137-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
138113

139114
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
140115

141116
RUN yum install -y \
142117
${NV_CUDNN_PACKAGE} \
143-
${NV_CUDNN_PACKAGE_DEV} \
144118
&& yum clean all \
145119
&& rm -rf /var/cache/yum/*
146120

0 commit comments

Comments
 (0)