Dockerfile 1.25 KB
Newer Older
Ankhi Karmakar IN's avatar
Ankhi Karmakar IN committed
1
2
3
4
5
6
7
8
9
10
11
ARG IMAGE_NAME
FROM ${IMAGE_NAME}:11.7.0-base-rockylinux8 as base

ENV NV_CUDA_LIB_VERSION 11.7.0-1

FROM base as base-amd64

ENV NV_NVTX_VERSION 11.7.50-1
ENV NV_LIBNPP_VERSION 11.7.3.21-1
ENV NV_LIBNPP_PACKAGE libnpp-11-7-${NV_LIBNPP_VERSION}
ENV NV_LIBCUBLAS_VERSION 11.10.1.25-1
12
13
14
15
16
ENV NV_LIBNCCL_PACKAGE_NAME libnccl
ENV NV_LIBNCCL_PACKAGE_VERSION 2.13.4-1
ENV NV_LIBNCCL_VERSION 2.13.4
ENV NCCL_VERSION 2.13.4
ENV NV_LIBNCCL_PACKAGE ${NV_LIBNCCL_PACKAGE_NAME}-${NV_LIBNCCL_PACKAGE_VERSION}+cuda11.7
Ankhi Karmakar IN's avatar
Ankhi Karmakar IN committed
17
18
19
20
21
22
FROM base as base-arm64

ENV NV_NVTX_VERSION 11.7.50-1
ENV NV_LIBNPP_VERSION 11.7.3.21-1
ENV NV_LIBNPP_PACKAGE libnpp-11-7-${NV_LIBNPP_VERSION}
ENV NV_LIBCUBLAS_VERSION 11.10.1.25-1
23
24
25
26
27
ENV NV_LIBNCCL_PACKAGE_NAME libnccl
ENV NV_LIBNCCL_PACKAGE_VERSION 2.12.12-1
ENV NV_LIBNCCL_VERSION 2.12.12
ENV NCCL_VERSION 2.12.12
ENV NV_LIBNCCL_PACKAGE ${NV_LIBNCCL_PACKAGE_NAME}-${NV_LIBNCCL_PACKAGE_VERSION}+cuda11.7
Ankhi Karmakar IN's avatar
Ankhi Karmakar IN committed
28
29
30
31
32
33
34
35
36
37
38
FROM base-${TARGETARCH}

ARG TARGETARCH

LABEL maintainer "NVIDIA CORPORATION <sw-cuda-installer@nvidia.com>"

RUN yum install -y \
    cuda-libraries-11-7-${NV_CUDA_LIB_VERSION} \
    cuda-nvtx-11-7-${NV_NVTX_VERSION} \
    ${NV_LIBNPP_PACKAGE} \
    libcublas-11-7-${NV_LIBCUBLAS_VERSION} \
39
    ${NV_LIBNCCL_PACKAGE} \
Ankhi Karmakar IN's avatar
Ankhi Karmakar IN committed
40
41
    && yum clean all \
    && rm -rf /var/cache/yum/*