Skip to content

Commit 3c5d2f5

Browse files
jiridanekjiridanek
authored
RHAIENG-568: Issue #385: decrease Python 3.11 CUDA images size by installing less (#1597)
1 parent 02e3b19 commit 3c5d2f5

File tree

7 files changed

+21
-204
lines changed

7 files changed

+21
-204
lines changed

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

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

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

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

133-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
134-
135-
# Install CUDA devel cudnn9 from:
136-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
111+
# Install CUDA cudnn9 from:
112+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
137113
ENV NV_CUDNN_VERSION=9.5.1.17-1
138114
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
139-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
140115

141116
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
142117

143118
RUN yum install -y \
144119
${NV_CUDNN_PACKAGE} \
145-
${NV_CUDNN_PACKAGE_DEV} \
146120
&& yum clean all \
147121
&& rm -rf /var/cache/yum/*
148122

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

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

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

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

146-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
147-
148-
# Install CUDA devel cudnn9 from:
149-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
124+
# Install CUDA cudnn9 from:
125+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
150126
ENV NV_CUDNN_VERSION=9.5.1.17-1
151127
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
152-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
153128

154129
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
155130

156131
RUN yum install -y \
157132
${NV_CUDNN_PACKAGE} \
158-
${NV_CUDNN_PACKAGE_DEV} \
159133
&& yum clean all \
160134
&& rm -rf /var/cache/yum/*
161135

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

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

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

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

146-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
147-
148-
# Install CUDA devel cudnn9 from:
149-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
124+
# Install CUDA cudnn9 from:
125+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
150126
ENV NV_CUDNN_VERSION=9.5.1.17-1
151127
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
152-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
153128

154129
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
155130

156131
RUN yum install -y \
157132
${NV_CUDNN_PACKAGE} \
158-
${NV_CUDNN_PACKAGE_DEV} \
159133
&& yum clean all \
160134
&& rm -rf /var/cache/yum/*
161135

‎rstudio/c9s-python-3.11/Dockerfile.cuda

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

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

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

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

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

149124
RUN yum install -y \
150125
${NV_CUDNN_PACKAGE} \
151-
${NV_CUDNN_PACKAGE_DEV} \
152126
&& yum clean all \
153127
&& rm -rf /var/cache/yum/*
154128

‎rstudio/rhel9-python-3.11/Dockerfile.cuda

Lines changed: 3 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -129,50 +129,23 @@ RUN yum install -y \
129129
# Set this flag so that libraries can find the location of CUDA
130130
ENV XLA_FLAGS=--xla_gpu_cuda_data_dir=/usr/local/cuda
131131

132-
# Install CUDA devel from:
133-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.4.1/ubi9/devel/Dockerfile
134-
ENV NV_CUDA_LIB_VERSION=12.4.1-1
135-
ENV NV_CUDA_CUDART_DEV_VERSION=12.4.127-1
136-
ENV NV_NVML_DEV_VERSION=12.4.127-1
137-
ENV NV_LIBCUBLAS_DEV_VERSION=12.4.5.8-1
138-
ENV NV_LIBNPP_DEV_VERSION=12.2.5.30-1
139-
ENV NV_LIBNPP_DEV_PACKAGE=libnpp-devel-12-4-${NV_LIBNPP_DEV_VERSION}
140-
ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-devel
141-
ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=2.21.5-1
142-
ENV NCCL_VERSION=2.21.5
143-
ENV NV_LIBNCCL_DEV_PACKAGE=${NV_LIBNCCL_DEV_PACKAGE_NAME}-${NV_LIBNCCL_DEV_PACKAGE_VERSION}+cuda12.4
144-
ENV NV_CUDA_NSIGHT_COMPUTE_VERSION=12.4.1-1
145-
ENV NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE=cuda-nsight-compute-12-4-${NV_CUDA_NSIGHT_COMPUTE_VERSION}
132+
# Install devel tools
146133

147134
RUN yum install -y \
148135
make \
149136
findutils \
150-
cuda-command-line-tools-12-4-${NV_CUDA_LIB_VERSION} \
151-
cuda-libraries-devel-12-4-${NV_CUDA_LIB_VERSION} \
152-
cuda-minimal-build-12-4-${NV_CUDA_LIB_VERSION} \
153-
cuda-cudart-devel-12-4-${NV_CUDA_CUDART_DEV_VERSION} \
154-
${NV_NVPROF_DEV_PACKAGE} \
155-
cuda-nvml-devel-12-4-${NV_NVML_DEV_VERSION} \
156-
libcublas-devel-12-4-${NV_LIBCUBLAS_DEV_VERSION} \
157-
${NV_LIBNPP_DEV_PACKAGE} \
158-
${NV_LIBNCCL_DEV_PACKAGE} \
159-
${NV_CUDA_NSIGHT_COMPUTE_DEV_PACKAGE} \
160137
&& yum clean all \
161138
&& rm -rf /var/cache/yum/*
162139

163-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
164-
165-
# Install CUDA devel cudnn8 from:
166-
# hhttps://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.4.1/ubi9/devel/cudnn/Dockerfile
140+
# Install CUDA cudnn8 from:
141+
# hhttps://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.4.1/ubi9/runtime/cudnn/Dockerfile
167142
ENV NV_CUDNN_VERSION=9.1.0.70-1
168143
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
169-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
170144

171145
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
172146

173147
RUN yum install -y \
174148
${NV_CUDNN_PACKAGE} \
175-
${NV_CUDNN_PACKAGE_DEV} \
176149
&& yum clean all \
177150
&& rm -rf /var/cache/yum/*
178151

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

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

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

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

133-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
134-
135-
# Install CUDA devel cudnn9 from:
136-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
111+
# Install CUDA cudnn9 from:
112+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
137113
ENV NV_CUDNN_VERSION=9.5.1.17-1
138114
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
139-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
140115

141116
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
142117

143118
RUN yum install -y \
144119
${NV_CUDNN_PACKAGE} \
145-
${NV_CUDNN_PACKAGE_DEV} \
146120
&& yum clean all \
147121
&& rm -rf /var/cache/yum/*
148122

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

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

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

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

133-
ENV LIBRARY_PATH=/usr/local/cuda/lib64/stubs
134-
135-
# Install CUDA devel cudnn9 from:
136-
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/devel/cudnn/Dockerfile
111+
# Install CUDA cudnn9 from:
112+
# https://gitlab.com/nvidia/container-images/cuda/-/blob/master/dist/12.6.3/ubi9/runtime/cudnn/Dockerfile
137113
ENV NV_CUDNN_VERSION=9.5.1.17-1
138114
ENV NV_CUDNN_PACKAGE=libcudnn9-cuda-12-${NV_CUDNN_VERSION}
139-
ENV NV_CUDNN_PACKAGE_DEV=libcudnn9-devel-cuda-12-${NV_CUDNN_VERSION}
140115

141116
LABEL com.nvidia.cudnn.version="${NV_CUDNN_VERSION}"
142117

143118
RUN yum install -y \
144119
${NV_CUDNN_PACKAGE} \
145-
${NV_CUDNN_PACKAGE_DEV} \
146120
&& yum clean all \
147121
&& rm -rf /var/cache/yum/*
148122

0 commit comments

Comments
 (0)