<reponame>eagleusb/docker-nvidia-cuda<filename>dist/11.5.0/centos8/devel/Dockerfile
ARG IMAGE_NAME
FROM ${IMAGE_NAME}:11.5.0-runtime-centos8 as base

FROM base as base-amd64

ENV NV_CUDA_LIB_VERSION 11.5.0-1
ENV NV_NVPROF_VERSION 11.5.50-1
ENV NV_CUDA_CUDART_DEV_VERSION 11.5.50-1
ENV NV_NVML_DEV_VERSION 11.5.50-1
ENV NV_LIBCUBLAS_DEV_VERSION 11.7.3.1-1
ENV NV_LIBNPP_DEV_VERSION 11.5.1.53-1
ENV NV_LIBNPP_DEV_PACKAGE libnpp-devel-11-5-${NV_LIBNPP_DEV_VERSION}

FROM base as base-arm64

ENV NV_CUDA_LIB_VERSION 11.5.0-1
ENV NV_NVPROF_VERSION 11.5.50-1
ENV NV_CUDA_CUDART_DEV_VERSION 11.5.50-1
ENV NV_NVML_DEV_VERSION 11.5.50-1
ENV NV_LIBCUBLAS_DEV_VERSION 11.7.3.1-1
ENV NV_LIBNPP_DEV_VERSION 11.5.1.53-1
ENV NV_LIBNPP_DEV_PACKAGE libnpp-devel-11-5-${NV_LIBNPP_DEV_VERSION}

FROM base as base-ppc64le

ENV NV_CUDA_LIB_VERSION 11.5.0-1
ENV NV_NVPROF_VERSION 11.5.50-1
ENV NV_CUDA_CUDART_DEV_VERSION 11.5.50-1
ENV NV_NVML_DEV_VERSION 11.5.50-1
ENV NV_LIBCUBLAS_DEV_VERSION 11.7.3.1-1
ENV NV_LIBNPP_DEV_VERSION 11.5.1.53-1
ENV NV_LIBNPP_DEV_PACKAGE libnpp-devel-11-5-${NV_LIBNPP_DEV_VERSION}

FROM base-${TARGETARCH}

ARG TARGETARCH

LABEL maintainer "NVIDIA CORPORATION <<EMAIL>>"

RUN yum install -y \
    make \
    cuda-command-line-tools-11-5-${NV_CUDA_LIB_VERSION} \
    cuda-libraries-devel-11-5-${NV_CUDA_LIB_VERSION} \
    cuda-minimal-build-11-5-${NV_CUDA_LIB_VERSION} \
    cuda-cudart-devel-11-5-${NV_CUDA_CUDART_DEV_VERSION} \
    cuda-nvprof-11-5-${NV_NVPROF_VERSION} \
    cuda-nvml-devel-11-5-${NV_NVML_DEV_VERSION} \
    libcublas-devel-11-5-${NV_LIBCUBLAS_DEV_VERSION} \
    ${NV_LIBNPP_DEV_PACKAGE} \
    && yum clean all \
    && rm -rf /var/cache/yum/*

ENV LIBRARY_PATH /usr/local/cuda/lib64/stubs