@@ -5,13 +5,6 @@ ARG CUDNN_CUDA_VERSION_MAJ_MIN
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
9- ARG NCCL_VERSION_AMD64=${NCCL_VERSION}
10- ARG NCCL_VERSION_ARM64=${NCCL_VERSION}
11- ARG NCCL_CUDA_VERSION_MAJ_MIN
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-
158ARG BUILD_ON_IMAGE
169ARG CUDNN_VERSION_MAJ=${CUDNN_VERSION%%.*}
1710ARG CUDNN_VERSION_MAJ=${CUDNN_VERSION_MAJ:-${CUDNN_VERSION_AMD64%%.*}}
@@ -23,47 +16,30 @@ FROM ${BUILD_ON_IMAGE} AS cudnn8-runtime-amd64
2316ARG CUDNN_VERSION_AMD64
2417ARG CUDNN_CUDA_VERSION_MAJ_MIN_AMD64
2518
26- ARG NCCL_VERSION_AMD64
27- ARG NCCL_CUDA_VERSION_MAJ_MIN_AMD64
28-
2919LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_AMD64}"
3020
3121ENV CUDNN_VERSION=${CUDNN_VERSION_AMD64}
3222ENV NV_CUDNN_PACKAGE_VERSION=${CUDNN_VERSION_AMD64}
3323ENV NV_CUDNN_PACKAGE_NAME=libcudnn${NV_CUDNN_PACKAGE_VERSION%%.*}
3424ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64}"
3525
36- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
37- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
38- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
39-
4026FROM ${BUILD_ON_IMAGE} AS cudnn8-runtime-arm64
4127
4228ARG CUDNN_VERSION_ARM64
4329ARG CUDNN_CUDA_VERSION_MAJ_MIN_ARM64
4430
45- ARG NCCL_VERSION_ARM64
46- ARG NCCL_CUDA_VERSION_MAJ_MIN_ARM64
47-
4831LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_ARM64}"
4932
5033ENV CUDNN_VERSION=${CUDNN_VERSION_ARM64}
5134ENV NV_CUDNN_PACKAGE_VERSION=${CUDNN_VERSION_ARM64}
5235ENV NV_CUDNN_PACKAGE_NAME=libcudnn${NV_CUDNN_PACKAGE_VERSION%%.*}
5336ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64}"
5437
55- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
56- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
57- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
58-
5938FROM ${BUILD_ON_IMAGE} AS cudnn8-devel-amd64
6039
6140ARG CUDNN_VERSION_AMD64
6241ARG CUDNN_CUDA_VERSION_MAJ_MIN_AMD64
6342
64- ARG NCCL_VERSION_AMD64
65- ARG NCCL_CUDA_VERSION_MAJ_MIN_AMD64
66-
6743LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_AMD64}"
6844
6945ENV CUDNN_VERSION=${CUDNN_VERSION_AMD64}
@@ -74,22 +50,11 @@ ENV NV_CUDNN_DEV_PACKAGE_NAME=libcudnn${NV_CUDNN_DEV_PACKAGE_VERSION%%.*}-dev
7450ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64}"
7551ENV NV_CUDNN_DEV_PACKAGE="${NV_CUDNN_DEV_PACKAGE_NAME}=${NV_CUDNN_DEV_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64}"
7652
77- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
78- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
79- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
80-
81- ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev
82- ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
83- ENV NV_LIBNCCL_DEV_PACKAGE="${NV_LIBNCCL_DEV_PACKAGE_NAME}=${NV_LIBNCCL_DEV_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
84-
8553FROM ${BUILD_ON_IMAGE} AS cudnn8-devel-arm64
8654
8755ARG CUDNN_VERSION_ARM64
8856ARG CUDNN_CUDA_VERSION_MAJ_MIN_ARM64
8957
90- ARG NCCL_VERSION_ARM64
91- ARG NCCL_CUDA_VERSION_MAJ_MIN_ARM64
92-
9358LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_ARM64}"
9459
9560ENV CUDNN_VERSION=${CUDNN_VERSION_ARM64}
@@ -100,60 +65,35 @@ ENV NV_CUDNN_DEV_PACKAGE_NAME=libcudnn${NV_CUDNN_DEV_PACKAGE_VERSION%%.*}-dev
10065ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64}"
10166ENV NV_CUDNN_DEV_PACKAGE="${NV_CUDNN_DEV_PACKAGE_NAME}=${NV_CUDNN_DEV_PACKAGE_VERSION}-1+cuda${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64}"
10267
103- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
104- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
105- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
106-
107- ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev
108- ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
109- ENV NV_LIBNCCL_DEV_PACKAGE="${NV_LIBNCCL_DEV_PACKAGE_NAME}=${NV_LIBNCCL_DEV_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
110-
11168FROM ${BUILD_ON_IMAGE} AS cudnn9-runtime-amd64
11269
11370ARG CUDNN_VERSION_AMD64
11471ARG CUDNN_CUDA_VERSION_MAJ_MIN_AMD64
11572
116- ARG NCCL_VERSION_AMD64
117- ARG NCCL_CUDA_VERSION_MAJ_MIN_AMD64
118-
11973LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_AMD64}"
12074
12175ENV CUDNN_VERSION=${CUDNN_VERSION_AMD64}
12276ENV NV_CUDNN_PACKAGE_VERSION=${CUDNN_VERSION_AMD64}
12377ENV NV_CUDNN_PACKAGE_NAME=libcudnn${NV_CUDNN_PACKAGE_VERSION%%.*}-cuda-${CUDNN_CUDA_VERSION_MAJ_MIN_AMD64%%.*}
12478ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1"
12579
126- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
127- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
128- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
129-
13080FROM ${BUILD_ON_IMAGE} AS cudnn9-runtime-arm64
13181
13282ARG CUDNN_VERSION_ARM64
13383ARG CUDNN_CUDA_VERSION_MAJ_MIN_ARM64
13484
135- ARG NCCL_VERSION_ARM64
136- ARG NCCL_CUDA_VERSION_MAJ_MIN_ARM64
137-
13885LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_ARM64}"
13986
14087ENV CUDNN_VERSION=${CUDNN_VERSION_ARM64}
14188ENV NV_CUDNN_PACKAGE_VERSION=${CUDNN_VERSION_ARM64}
14289ENV NV_CUDNN_PACKAGE_NAME=libcudnn${NV_CUDNN_PACKAGE_VERSION%%.*}-cuda-${CUDNN_CUDA_VERSION_MAJ_MIN_ARM64%%.*}
14390ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1"
14491
145- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
146- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
147- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
148-
14992FROM ${BUILD_ON_IMAGE} AS cudnn9-devel-amd64
15093
15194ARG CUDNN_VERSION_AMD64
15295ARG CUDNN_CUDA_VERSION_MAJ_MIN_AMD64
15396
154- ARG NCCL_VERSION_AMD64
155- ARG NCCL_CUDA_VERSION_MAJ_MIN_AMD64
156-
15797LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_AMD64}"
15898
15999ENV CUDNN_VERSION=${CUDNN_VERSION_AMD64}
@@ -167,22 +107,11 @@ ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1"
167107ENV NV_CUDNN_DEV_PACKAGE="${NV_CUDNN_DEV_PACKAGE_NAME}=${NV_CUDNN_DEV_PACKAGE_VERSION}-1"
168108ENV NV_CUDNN_HEADERS_PACKAGE="${NV_CUDNN_HEADERS_PACKAGE_NAME}=${NV_CUDNN_HEADERS_PACKAGE_VERSION}-1"
169109
170- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
171- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
172- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
173-
174- ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev
175- ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=${NCCL_VERSION_AMD64}
176- ENV NV_LIBNCCL_DEV_PACKAGE="${NV_LIBNCCL_DEV_PACKAGE_NAME}=${NV_LIBNCCL_DEV_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_AMD64}"
177-
178110FROM ${BUILD_ON_IMAGE} AS cudnn9-devel-arm64
179111
180112ARG CUDNN_VERSION_ARM64
181113ARG CUDNN_CUDA_VERSION_MAJ_MIN_ARM64
182114
183- ARG NCCL_VERSION_ARM64
184- ARG NCCL_CUDA_VERSION_MAJ_MIN_ARM64
185-
186115LABEL com.nvidia.cudnn.version="${CUDNN_VERSION_ARM64}"
187116
188117ENV CUDNN_VERSION=${CUDNN_VERSION_ARM64}
@@ -196,14 +125,6 @@ ENV NV_CUDNN_PACKAGE="${NV_CUDNN_PACKAGE_NAME}=${NV_CUDNN_PACKAGE_VERSION}-1"
196125ENV NV_CUDNN_DEV_PACKAGE="${NV_CUDNN_DEV_PACKAGE_NAME}=${NV_CUDNN_DEV_PACKAGE_VERSION}-1"
197126ENV NV_CUDNN_HEADERS_PACKAGE="${NV_CUDNN_HEADERS_PACKAGE_NAME}=${NV_CUDNN_HEADERS_PACKAGE_VERSION}-1"
198127
199- ENV NV_LIBNCCL_PACKAGE_NAME=libnccl2
200- ENV NV_LIBNCCL_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
201- ENV NV_LIBNCCL_PACKAGE="${NV_LIBNCCL_PACKAGE_NAME}=${NV_LIBNCCL_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
202-
203- ENV NV_LIBNCCL_DEV_PACKAGE_NAME=libnccl-dev
204- ENV NV_LIBNCCL_DEV_PACKAGE_VERSION=${NCCL_VERSION_ARM64}
205- ENV NV_LIBNCCL_DEV_PACKAGE="${NV_LIBNCCL_DEV_PACKAGE_NAME}=${NV_LIBNCCL_DEV_PACKAGE_VERSION}-1+cuda${NCCL_CUDA_VERSION_MAJ_MIN_ARM64}"
206-
207128FROM cudnn${CUDNN_VERSION_MAJ}-${CUDA_IMAGE_FLAVOR}-${TARGETARCH}
208129
209130ARG DEBIAN_FRONTEND=noninteractive
@@ -255,13 +176,5 @@ RUN cpuBlasLib="$(update-alternatives --query \
255176 ${NV_CUDNN_PACKAGE_NAME} \
256177 ${NV_CUDNN_DEV_PACKAGE_NAME} \
257178 ${NV_CUDNN_HEADERS_PACKAGE_NAME} \
258- # # Install NCCL
259- && apt-get -y install --no-install-recommends \
260- ${NV_LIBNCCL_PACKAGE} \
261- ${NV_LIBNCCL_DEV_PACKAGE} \
262- # # Keep apt from auto upgrading the NCCL package
263- && apt-mark hold \
264- ${NV_LIBNCCL_PACKAGE_NAME} \
265- ${NV_LIBNCCL_DEV_PACKAGE_NAME} \
266179 # # Clean up
267180 && rm -rf /var/lib/apt/lists/*
0 commit comments