@@ -5,16 +5,9 @@ ARG CUDNN_CUDA_VERSION_MAJ_MIN=13.4
55ARG CUDNN_CUDA_VERSION_MAJ_MIN_AMD64=${CUDNN_CUDA_VERSION_MAJ_MIN}
66ARG CUDNN_CUDA_VERSION_MAJ_MIN_ARM64=${CUDNN_CUDA_VERSION_MAJ_MIN}
77
8- ARG NCCL_VERSION=2.31.2
9- ARG NCCL_VERSION_AMD64=${NCCL_VERSION}
10- ARG NCCL_VERSION_ARM64=${NCCL_VERSION}
11- ARG NCCL_CUDA_VERSION_MAJ_MIN=13.4
12- ARG NCCL_CUDA_VERSION_MAJ_MIN_AMD64=${NCCL_CUDA_VERSION_MAJ_MIN}
13- ARG NCCL_CUDA_VERSION_MAJ_MIN_ARM64=${NCCL_CUDA_VERSION_MAJ_MIN}
14-
158# # CUDA_IMAGE
16- # # ├── nvidia/cuda:13.4.2 -runtime-ubuntu24.04
17- # # └── nvidia/cuda:13.4.2 -devel-ubuntu24.04
9+ # # ├── nvidia/cuda:13.4.3 -runtime-ubuntu24.04
10+ # # └── nvidia/cuda:13.4.3 -devel-ubuntu24.04
1811ARG BUILD_ON_IMAGE
1912ARG CUDNN_VERSION_MAJ=${CUDNN_VERSION%%.*}
2013ARG CUDNN_VERSION_MAJ=${CUDNN_VERSION_MAJ:-${CUDNN_VERSION_AMD64%%.*}}
@@ -26,47 +19,30 @@ FROM ${BUILD_ON_IMAGE} AS cudnn8-runtime-amd64
2619ARG CUDNN_VERSION_AMD64
2720ARG CUDNN_CUDA_VERSION_MAJ_MIN_AMD64
2821
29- ARG NCCL_VERSION_AMD64
30- ARG NCCL_CUDA_VERSION_MAJ_MIN_AMD64
31-
3222LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_AMD64}"
3323
3424ENV CUDNN_VERSION=${CUDNN_VERSION_AMD64}
3525ENV NV_CUDNN_PACKAGE_VERSION=${CUDNN_VERSION_AMD64}
3626ENV NV_CUDNN_PACKAGE_NAME=libcudnn${NV_CUDNN_PACKAGE_VERSION%%.*}
3727ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64}"
3828
39- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
40- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
41- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
42-
4329FROM ${BUILD_ON_IMAGE} AS cudnn8-runtime-arm64
4430
4531ARG CUDNN_VERSION_ARM64
4632ARG CUDNN_CUDA_VERSION_MAJ_MIN_ARM64
4733
48- ARG NCCL_VERSION_ARM64
49- ARG NCCL_CUDA_VERSION_MAJ_MIN_ARM64
50-
5134LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_ARM64}"
5235
5336ENV CUDNN_VERSION=${CUDNN_VERSION_ARM64}
5437ENV NV_CUDNN_PACKAGE_VERSION=${CUDNN_VERSION_ARM64}
5538ENV NV_CUDNN_PACKAGE_NAME=libcudnn${NV_CUDNN_PACKAGE_VERSION%%.*}
5639ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64}"
5740
58- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
59- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
60- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
61-
6241FROM ${BUILD_ON_IMAGE} AS cudnn8-devel-amd64
6342
6443ARG CUDNN_VERSION_AMD64
6544ARG CUDNN_CUDA_VERSION_MAJ_MIN_AMD64
6645
67- ARG NCCL_VERSION_AMD64
68- ARG NCCL_CUDA_VERSION_MAJ_MIN_AMD64
69-
7046LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_AMD64}"
7147
7248ENV CUDNN_VERSION=${CUDNN_VERSION_AMD64}
@@ -77,22 +53,11 @@ ENV NV_CUDNN_DEV_PACKAGE_NAME=libcudnn${NV_CUDNN_DEV_PACKAGE_VERSION%%.*}-dev
7753ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64}"
7854ENV NV_CUDNN_DEV_PACKAGE="${NV_CUDNN_DEV_PACKAGE_NAME}=${NV_CUDNN_DEV_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64}"
7955
80- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
81- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
82- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
83-
84- ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev
85- ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
86- ENV NV_LIBNCCL_DEV_PACKAGE="${NV_LIBNCCL_DEV_PACKAGE_NAME}=${NV_LIBNCCL_DEV_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
87-
8856FROM ${BUILD_ON_IMAGE} AS cudnn8-devel-arm64
8957
9058ARG CUDNN_VERSION_ARM64
9159ARG CUDNN_CUDA_VERSION_MAJ_MIN_ARM64
9260
93- ARG NCCL_VERSION_ARM64
94- ARG NCCL_CUDA_VERSION_MAJ_MIN_ARM64
95-
9661LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_ARM64}"
9762
9863ENV CUDNN_VERSION=${CUDNN_VERSION_ARM64}
@@ -103,60 +68,35 @@ ENV NV_CUDNN_DEV_PACKAGE_NAME=libcudnn${NV_CUDNN_DEV_PACKAGE_VERSION%%.*}-dev
10368ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64}"
10469ENV NV_CUDNN_DEV_PACKAGE="${NV_CUDNN_DEV_PACKAGE_NAME}=${NV_CUDNN_DEV_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64}"
10570
106- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
107- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
108- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
109-
110- ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev
111- ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
112- ENV NV_LIBNCCL_DEV_PACKAGE="${NV_LIBNCCL_DEV_PACKAGE_NAME}=${NV_LIBNCCL_DEV_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
113-
11471FROM ${BUILD_ON_IMAGE} AS cudnn9-runtime-amd64
11572
11673ARG CUDNN_VERSION_AMD64
11774ARG CUDNN_CUDA_VERSION_MAJ_MIN_AMD64
11875
119- ARG NCCL_VERSION_AMD64
120- ARG NCCL_CUDA_VERSION_MAJ_MIN_AMD64
121-
12276LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_AMD64}"
12377
12478ENV CUDNN_VERSION=${CUDNN_VERSION_AMD64}
12579ENV NV_CUDNN_PACKAGE_VERSION=${CUDNN_VERSION_AMD64}
12680ENV NV_CUDNN_PACKAGE_NAME=libcudnn${NV_CUDNN_PACKAGE_VERSION%%.*}-cuda-${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64%%.*}
12781ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1"
12882
129- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
130- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
131- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
132-
13383FROM ${BUILD_ON_IMAGE} AS cudnn9-runtime-arm64
13484
13585ARG CUDNN_VERSION_ARM64
13686ARG CUDNN_CUDA_VERSION_MAJ_MIN_ARM64
13787
138- ARG NCCL_VERSION_ARM64
139- ARG NCCL_CUDA_VERSION_MAJ_MIN_ARM64
140-
14188LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_ARM64}"
14289
14390ENV CUDNN_VERSION=${CUDNN_VERSION_ARM64}
14491ENV NV_CUDNN_PACKAGE_VERSION=${CUDNN_VERSION_ARM64}
14592ENV NV_CUDNN_PACKAGE_NAME=libcudnn${NV_CUDNN_PACKAGE_VERSION%%.*}-cuda-${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64%%.*}
14693ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1"
14794
148- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
149- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
150- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
151-
15295FROM ${BUILD_ON_IMAGE} AS cudnn9-devel-amd64
15396
15497ARG CUDNN_VERSION_AMD64
15598ARG CUDNN_CUDA_VERSION_MAJ_MIN_AMD64
15699
157- ARG NCCL_VERSION_AMD64
158- ARG NCCL_CUDA_VERSION_MAJ_MIN_AMD64
159-
160100LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_AMD64}"
161101
162102ENV CUDNN_VERSION=${CUDNN_VERSION_AMD64}
@@ -170,22 +110,11 @@ ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1"
170110ENV NV_CUDNN_DEV_PACKAGE="${NV_CUDNN_DEV_PACKAGE_NAME}=${NV_CUDNN_DEV_PACKAGE_VERSION}-1"
171111ENV NV_CUDNN_HEADERS_PACKAGE="${NV_CUDNN_HEADERS_PACKAGE_NAME}=${NV_CUDNN_HEADERS_PACKAGE_VERSION}-1"
172112
173- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
174- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
175- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
176-
177- ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev
178- ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
179- ENV NV_LIBNCCL_DEV_PACKAGE="${NV_LIBNCCL_DEV_PACKAGE_NAME}=${NV_LIBNCCL_DEV_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
180-
181113FROM ${BUILD_ON_IMAGE} AS cudnn9-devel-arm64
182114
183115ARG CUDNN_VERSION_ARM64
184116ARG CUDNN_CUDA_VERSION_MAJ_MIN_ARM64
185117
186- ARG NCCL_VERSION_ARM64
187- ARG NCCL_CUDA_VERSION_MAJ_MIN_ARM64
188-
189118LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_ARM64}"
190119
191120ENV CUDNN_VERSION=${CUDNN_VERSION_ARM64}
@@ -199,14 +128,6 @@ ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1"
199128ENV NV_CUDNN_DEV_PACKAGE="${NV_CUDNN_DEV_PACKAGE_NAME}=${NV_CUDNN_DEV_PACKAGE_VERSION}-1"
200129ENV NV_CUDNN_HEADERS_PACKAGE="${NV_CUDNN_HEADERS_PACKAGE_NAME}=${NV_CUDNN_HEADERS_PACKAGE_VERSION}-1"
201130
202- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
203- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
204- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
205-
206- ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev
207- ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
208- ENV NV_LIBNCCL_DEV_PACKAGE="${NV_LIBNCCL_DEV_PACKAGE_NAME}=${NV_LIBNCCL_DEV_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
209-
210131FROM cudnn${CUDNN_VERSION_MAJ}-${CUDA_IMAGE_FLAVOR}-${TARGETARCH}
211132
212133ARG DEBIAN_FRONTEND=noninteractive
@@ -258,13 +179,5 @@ RUN cpuBlasLib="$(update-alternatives --query \
258179 ${NV_CUDNN_PACKAGE_NAME} \
259180 ${NV_CUDNN_DEV_PACKAGE_NAME} \
260181 ${NV_CUDNN_HEADERS_PACKAGE_NAME} \
261- # # Install NCCL
262- && apt-get -y install --no-install-recommends \
263- ${NV_LIBNCCL_PACKAGE} \
264- ${NV_LIBNCCL_DEV_PACKAGE} \
265- # # Keep apt from auto upgrading the NCCL package
266- && apt-mark hold \
267- ${NV_LIBNCCL_PACKAGE_NAME} \
268- ${NV_LIBNCCL_DEV_PACKAGE_NAME} \
269182 # # Clean up
270183 && rm -rf /var/lib/apt/lists/*
0 commit comments