Skip to content

Commit e8eb7ff

Browse files
jiridanekjiridanek
authored
RHAIENG-568: Issue opendatahub-io#385: decrease Python 3.11 CUDA images size by installing less (opendatahub-io#1597) (#1491)
* opendatahub-io#1597
1 parent 0907c35 commit e8eb7ff

File tree

5 files changed

+15
-147
lines changed

5 files changed

+15
-147
lines changed

‎jupyter/minimal/ubi9-python-3.11/Dockerfile.konflux.cuda

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

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

122110
RUN yum install -y \
123111
make \
124112
findutils \
125-
cuda-command-line-tools-12-6-${NV_CUDA_LIB_VERSION} \
126-
cuda-libraries-devel-12-6-${NV_CUDA_LIB_VERSION} \
127-
cuda-minimal-build-12-6-${NV_CUDA_LIB_VERSION} \
128-
cuda-cudart-devel-12-6-${NV_CUDA_CUDART_DEV_VERSION} \
129-
${NV_NVPROF_DEV_PACKAGE} \
130-
cuda-nvml-devel-12-6-${NV_NVML_DEV_VERSION} \
131-
libcublas-devel-12-6-${NV_LIBCUBLAS_DEV_VERSION} \
132-
${NV_LIBNPP_DEV_PACKAGE} \
133-
${NV_LIBNCCL_DEV_PACKAGE} \
134-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
135113
&& yum clean all \
136114
&& rm -rf /var/cache/yum/*
137115

138-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
139-
140-
# Install CUDA devel cudnn9 from:
141-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
116+
# Install CUDA cudnn9 from:
117+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
142118
ENV NV_CUDNN_VERSION=9.5.1.17-1
143119
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
144-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
145120

146121
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
147122

148123
RUN yum install -y \
149124
${NV_CUDNN_PACKAGE} \
150-
${NV_CUDNN_PACKAGE_DEV} \
151125
&& yum clean all \
152126
&& rm -rf /var/cache/yum/*
153127

‎jupyter/pytorch/ubi9-python-3.11/Dockerfile.konflux.cuda

Lines changed: 3 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -118,49 +118,23 @@ RUN yum install -y \
118118
&& yum clean all \
119119
&& rm -rf /var/cache/yum/*
120120

121-
# Install CUDA devel from:
122-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/Dockerfile
123-
ENV NV_CUDA_CUDART_DEV_VERSION=12.6.77-1
124-
ENV NV_NVML_DEV_VERSION=12.6.77-1
125-
ENV NV_LIBCUBLAS_DEV_VERSION=12.6.4.1-1
126-
ENV NV_LIBNPP_DEV_VERSION=12.3.1.54-1
127-
ENV NV_LIBNPP_DEV_PACKAGE=libnpp-devel-12-6-${NV_LIBNPP_DEV_VERSION}
128-
ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-devel
129-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.23.4-1
130-
ENV NCCL_VERSION=2.23.4
131-
ENV NV_LIBNCCL_DEV_PACKAGE=${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.6
132-
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.6.3-1
133-
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-6-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
121+
# Install devel tools
134122

135123
RUN yum install -y \
136124
make \
137125
findutils \
138-
cuda-command-line-tools-12-6-${NV_CUDA_LIB_VERSION} \
139-
cuda-libraries-devel-12-6-${NV_CUDA_LIB_VERSION} \
140-
cuda-minimal-build-12-6-${NV_CUDA_LIB_VERSION} \
141-
cuda-cudart-devel-12-6-${NV_CUDA_CUDART_DEV_VERSION} \
142-
${NV_NVPROF_DEV_PACKAGE} \
143-
cuda-nvml-devel-12-6-${NV_NVML_DEV_VERSION} \
144-
libcublas-devel-12-6-${NV_LIBCUBLAS_DEV_VERSION} \
145-
${NV_LIBNPP_DEV_PACKAGE} \
146-
${NV_LIBNCCL_DEV_PACKAGE} \
147-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
148126
&& yum clean all \
149127
&& rm -rf /var/cache/yum/*
150128

151-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
152-
153-
# Install CUDA devel cudnn9 from:
154-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
129+
# Install CUDA cudnn9 from:
130+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
155131
ENV NV_CUDNN_VERSION=9.5.1.17-1
156132
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
157-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
158133

159134
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
160135

161136
RUN yum install -y \
162137
${NV_CUDNN_PACKAGE} \
163-
${NV_CUDNN_PACKAGE_DEV} \
164138
&& yum clean all \
165139
&& rm -rf /var/cache/yum/*
166140

@@ -184,7 +158,6 @@ WORKDIR /opt/app-root/bin
184158
COPY ${JUPYTER_REUSABLE_UTILS} utils/
185159

186160
COPY ${MINIMAL_SOURCE_CODE}/start-notebook.sh ./
187-
188161
USER 0
189162

190163
# Dependencies for PDF export begin

‎jupyter/tensorflow/ubi9-python-3.11/Dockerfile.konflux.cuda

Lines changed: 3 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -118,49 +118,23 @@ RUN yum install -y \
118118
&& yum clean all \
119119
&& rm -rf /var/cache/yum/*
120120

121-
# Install CUDA devel from:
122-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/Dockerfile
123-
ENV NV_CUDA_CUDART_DEV_VERSION=12.6.77-1
124-
ENV NV_NVML_DEV_VERSION=12.6.77-1
125-
ENV NV_LIBCUBLAS_DEV_VERSION=12.6.4.1-1
126-
ENV NV_LIBNPP_DEV_VERSION=12.3.1.54-1
127-
ENV NV_LIBNPP_DEV_PACKAGE=libnpp-devel-12-6-${NV_LIBNPP_DEV_VERSION}
128-
ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-devel
129-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.23.4-1
130-
ENV NCCL_VERSION=2.23.4
131-
ENV NV_LIBNCCL_DEV_PACKAGE=${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.6
132-
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.6.3-1
133-
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-6-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
121+
# Install devel tools
134122

135123
RUN yum install -y \
136124
make \
137125
findutils \
138-
cuda-command-line-tools-12-6-${NV_CUDA_LIB_VERSION} \
139-
cuda-libraries-devel-12-6-${NV_CUDA_LIB_VERSION} \
140-
cuda-minimal-build-12-6-${NV_CUDA_LIB_VERSION} \
141-
cuda-cudart-devel-12-6-${NV_CUDA_CUDART_DEV_VERSION} \
142-
${NV_NVPROF_DEV_PACKAGE} \
143-
cuda-nvml-devel-12-6-${NV_NVML_DEV_VERSION} \
144-
libcublas-devel-12-6-${NV_LIBCUBLAS_DEV_VERSION} \
145-
${NV_LIBNPP_DEV_PACKAGE} \
146-
${NV_LIBNCCL_DEV_PACKAGE} \
147-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
148126
&& yum clean all \
149127
&& rm -rf /var/cache/yum/*
150128

151-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
152-
153-
# Install CUDA devel cudnn9 from:
154-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
129+
# Install CUDA cudnn9 from:
130+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
155131
ENV NV_CUDNN_VERSION=9.5.1.17-1
156132
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
157-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
158133

159134
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
160135

161136
RUN yum install -y \
162137
${NV_CUDNN_PACKAGE} \
163-
${NV_CUDNN_PACKAGE_DEV} \
164138
&& yum clean all \
165139
&& rm -rf /var/cache/yum/*
166140

@@ -184,7 +158,6 @@ WORKDIR /opt/app-root/bin
184158
COPY ${JUPYTER_REUSABLE_UTILS} utils/
185159

186160
COPY ${MINIMAL_SOURCE_CODE}/start-notebook.sh ./
187-
188161
USER 0
189162

190163
# Dependencies for PDF export begin

‎runtimes/pytorch/ubi9-python-3.11/Dockerfile.konflux.cuda

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

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

122110
RUN yum install -y \
123111
make \
124112
findutils \
125-
cuda-command-line-tools-12-6-${NV_CUDA_LIB_VERSION} \
126-
cuda-libraries-devel-12-6-${NV_CUDA_LIB_VERSION} \
127-
cuda-minimal-build-12-6-${NV_CUDA_LIB_VERSION} \
128-
cuda-cudart-devel-12-6-${NV_CUDA_CUDART_DEV_VERSION} \
129-
${NV_NVPROF_DEV_PACKAGE} \
130-
cuda-nvml-devel-12-6-${NV_NVML_DEV_VERSION} \
131-
libcublas-devel-12-6-${NV_LIBCUBLAS_DEV_VERSION} \
132-
${NV_LIBNPP_DEV_PACKAGE} \
133-
${NV_LIBNCCL_DEV_PACKAGE} \
134-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
135113
&& yum clean all \
136114
&& rm -rf /var/cache/yum/*
137115

138-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
139-
140-
# Install CUDA devel cudnn9 from:
141-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
116+
# Install CUDA cudnn9 from:
117+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
142118
ENV NV_CUDNN_VERSION=9.5.1.17-1
143119
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
144-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
145120

146121
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
147122

148123
RUN yum install -y \
149124
${NV_CUDNN_PACKAGE} \
150-
${NV_CUDNN_PACKAGE_DEV} \
151125
&& yum clean all \
152126
&& rm -rf /var/cache/yum/*
153127

‎runtimes/tensorflow/ubi9-python-3.11/Dockerfile.konflux.cuda

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

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

122110
RUN yum install -y \
123111
make \
124112
findutils \
125-
cuda-command-line-tools-12-6-${NV_CUDA_LIB_VERSION} \
126-
cuda-libraries-devel-12-6-${NV_CUDA_LIB_VERSION} \
127-
cuda-minimal-build-12-6-${NV_CUDA_LIB_VERSION} \
128-
cuda-cudart-devel-12-6-${NV_CUDA_CUDART_DEV_VERSION} \
129-
${NV_NVPROF_DEV_PACKAGE} \
130-
cuda-nvml-devel-12-6-${NV_NVML_DEV_VERSION} \
131-
libcublas-devel-12-6-${NV_LIBCUBLAS_DEV_VERSION} \
132-
${NV_LIBNPP_DEV_PACKAGE} \
133-
${NV_LIBNCCL_DEV_PACKAGE} \
134-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
135113
&& yum clean all \
136114
&& rm -rf /var/cache/yum/*
137115

138-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
139-
140-
# Install CUDA devel cudnn9 from:
141-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
116+
# Install CUDA cudnn9 from:
117+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
142118
ENV NV_CUDNN_VERSION=9.5.1.17-1
143119
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
144-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
145120

146121
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
147122

148123
RUN yum install -y \
149124
${NV_CUDNN_PACKAGE} \
150-
${NV_CUDNN_PACKAGE_DEV} \
151125
&& yum clean all \
152126
&& rm -rf /var/cache/yum/*
153127

0 commit comments

Comments
 (0)