NVIDIA
Merlin PyTorch Training
Container
NVIDIA
Merlin PyTorch Training

This container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with PyTorch.

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
    ["/bin/bash" "-c" "/opt/nvidia/nvidia_entrypoint.sh"]
    05/06/2022 7:06 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
    /bin/bash
    05/06/2022 7:06 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4HEALTHCHECK
    &{["NONE"] "0s" "0s" "0s" '\x00'}
    05/06/2022 7:06 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=main MODELS_VER=main NVTAB_VER=main SYSTEMS_VER=main TF4REC_VER=main rm -rf /opt/conda/share/jupyter/lab/staging/node_modules/node-fetch
    05/06/2022 7:06 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=main MODELS_VER=main NVTAB_VER=main SYSTEMS_VER=main TF4REC_VER=main rm -rf /opt/conda/share/jupyter/lab/staging/node_modules/marked
    05/06/2022 7:06 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=main MODELS_VER=main NVTAB_VER=main SYSTEMS_VER=main TF4REC_VER=main rm -rf /repos
    05/06/2022 7:06 AM UTC
    sha256:8ca8e35c14d8259160eb3151d353420ec781e733cf3cbff5459a2660490c0fa5RUN
    CORE_VER=main MODELS_VER=main NVTAB_VER=main SYSTEMS_VER=main TF4REC_VER=main git clone https://github.com/NVIDIA-Merlin/Merlin/ /Merlin
    05/06/2022 7:06 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PYTHONPATH=:/core:/systems:/nvtabular:/transformers4rec:/models
    05/06/2022 7:06 AM UTC
    sha256:c614450efcc0d42c2505c5c8557cf00c9b31ca3b3039e806a305192d5a0e72e9RUN
    CORE_VER=main MODELS_VER=main NVTAB_VER=main SYSTEMS_VER=main TF4REC_VER=main git clone https://github.com/NVIDIA-Merlin/Models.git /models/ &&
      cd /models/ &&
      git checkout ${MODELS_VER} &&
      pip install . --no-deps
    05/06/2022 7:06 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PYTHONPATH=:/core:/systems:/nvtabular:/transformers4rec
    05/06/2022 7:05 AM UTC
    ...