Dockerfile.train

# Copyright (c) 2021 NVIDIA Corporation. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ==============================================================================


FROM nvcr.io/nvidia/merlin/merlin-training:0.5.1@sha256:ee00f2698f3453ae9f34eafb852524e42c439596c67d13fe959dc5fe159c90d5


RUN apt-get update && \
    apt-get install -y python3-pip vim curl ca-certificates gnupg

RUN curl -fsSLo /tmp/helm.tar.gz https://get.helm.sh/helm-v3.18.6-linux-amd64.tar.gz && \
    echo "3f43c0aa57243852dd542493a0f54f1396c0bc8ec7296bbb2c01e802010819ce  /tmp/helm.tar.gz" | sha256sum -c - && \
    tar -xzf /tmp/helm.tar.gz -C /tmp && \
    install -m 0755 /tmp/linux-amd64/helm /usr/local/bin/helm && \
    rm -rf /tmp/helm.tar.gz /tmp/linux-amd64

RUN curl -fsSLo /tmp/cloud.google.gpg https://packages.cloud.google.com/apt/doc/apt-key.gpg && \
    gpg --batch --yes --dearmor --output /usr/share/keyrings/cloud.google.gpg /tmp/cloud.google.gpg && \
    rm /tmp/cloud.google.gpg && \
    echo "deb [signed-by=/usr/share/keyrings/cloud.google.gpg] https://packages.cloud.google.com/apt cloud-sdk main" > /etc/apt/sources.list.d/google-cloud-sdk.list && \
    apt-get update && \
    apt-get install -y --no-install-recommends google-cloud-cli=579.0.0-0 && \
    rm -rf /var/lib/apt/lists/*


COPY preprocess-train /script

WORKDIR /script