ARG IMAGE_NAME
FROM ${IMAGE_NAME}:10.0-devel-centos7 as base

FROM base as base-amd64

ENV NV_CUDNN_VERSION 7.6.5.32-1
ENV NV_CUDNN_PACKAGE libcudnn7-${NV_CUDNN_VERSION}.cuda10.0
ENV NV_CUDNN_PACKAGE_DEV libcudnn7-devel-${NV_CUDNN_VERSION}.cuda10.0

ENV NV_CUDNN_DL_HASHCMD sha256sum
ENV NV_CUDNN_DL_SUM ced1d54b301053102d665716aaac7b74a9627eac7180f2d22f88c6437810d3b5
ENV NV_CUDNN_DL_BASENAME libcudnn7-7.6.5.32-1.cuda10.0.x86_64.rpm
ENV NV_CUDNN_DL_URL https://developer.download.nvidia.com/compute/machine-learning/repos/rhel7/x86_64/libcudnn7-7.6.5.32-1.cuda10.0.x86_64.rpm

RUN yum -y install wget

RUN wget -q ${NV_CUDNN_DL_URL} \
    && echo "${NV_CUDNN_DL_SUM}  ${NV_CUDNN_DL_BASENAME}" | ${NV_CUDNN_DL_HASHCMD} -c - \
    && rpm -i ${NV_CUDNN_DL_BASENAME} \
    && rm -f ${NV_CUDNN_DL_BASENAME}

ENV NV_CUDNN_DEV_DL_HASHCMD sha256sum
ENV NV_CUDNN_DEV_DL_SUM bd024f2629b1c4642662d8c99b46b078ac445ffa8b1c4c83ac8f466fd8a7d9d6
ENV NV_CUDNN_DEV_DL_BASENAME libcudnn7-devel-7.6.5.32-1.cuda10.0.x86_64.rpm
ENV NV_CUDNN_DEV_DL_URL https://developer.download.nvidia.com/compute/machine-learning/repos/rhel7/x86_64/libcudnn7-devel-7.6.5.32-1.cuda10.0.x86_64.rpm

RUN wget -q ${NV_CUDNN_DEV_DL_URL} \
    && echo "${NV_CUDNN_DEV_DL_SUM}  ${NV_CUDNN_DEV_DL_BASENAME}" | ${NV_CUDNN_DEV_DL_HASHCMD} -c - \
    && rpm -i ${NV_CUDNN_DEV_DL_BASENAME} \
    && rm -f ${NV_CUDNN_DEV_DL_BASENAME} \
    && yum -y remove wget


FROM base as base-ppc64le

ENV NV_CUDNN_VERSION 7.6.5.32-1
ENV NV_CUDNN_PACKAGE libcudnn7-${NV_CUDNN_VERSION}.cuda10.0
ENV NV_CUDNN_PACKAGE_DEV libcudnn7-devel-${NV_CUDNN_VERSION}.cuda10.0

ENV NV_CUDNN_DL_HASHCMD sha256sum
ENV NV_CUDNN_DL_SUM 64cfdc1fddbc0198ad99f9d109c56f3d15dd8882eb73c603b288d781de96c39a
ENV NV_CUDNN_DL_BASENAME libcudnn7-7.6.5.32-1.cuda10.0.ppc64le.rpm
ENV NV_CUDNN_DL_URL https://developer.download.nvidia.com/compute/machine-learning/repos/rhel7/ppc64le/libcudnn7-7.6.5.32-1.cuda10.0.ppc64le.rpm

RUN yum -y install wget

RUN wget -q ${NV_CUDNN_DL_URL} \
    && echo "${NV_CUDNN_DL_SUM}  ${NV_CUDNN_DL_BASENAME}" | ${NV_CUDNN_DL_HASHCMD} -c - \
    && rpm -i ${NV_CUDNN_DL_BASENAME} \
    && rm -f ${NV_CUDNN_DL_BASENAME}

ENV NV_CUDNN_DEV_DL_HASHCMD sha256sum
ENV NV_CUDNN_DEV_DL_SUM ce9f3d4f109f7c96a71a401ec7bd0acb869c80a7e0f3fda2bd58f4059a4bda29
ENV NV_CUDNN_DEV_DL_BASENAME libcudnn7-devel-7.6.5.32-1.cuda10.0.ppc64le.rpm
ENV NV_CUDNN_DEV_DL_URL https://developer.download.nvidia.com/compute/machine-learning/repos/rhel7/ppc64le/libcudnn7-devel-7.6.5.32-1.cuda10.0.ppc64le.rpm

RUN wget -q ${NV_CUDNN_DEV_DL_URL} \
    && echo "${NV_CUDNN_DEV_DL_SUM}  ${NV_CUDNN_DEV_DL_BASENAME}" | ${NV_CUDNN_DEV_DL_HASHCMD} -c - \
    && rpm -i ${NV_CUDNN_DEV_DL_BASENAME} \
    && rm -f ${NV_CUDNN_DEV_DL_BASENAME} \
    && yum -y remove wget


FROM base-${TARGETARCH}

ARG TARGETARCH

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

ENV CUDNN_VERSION ${NV_CUDNN_VERSION}
LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}"

RUN yum install -y \
    ${NV_CUDNN_PACKAGE} \
    ${NV_CUDNN_PACKAGE_DEV} \
    && yum clean all \
    && rm -rf /var/cache/yum/*