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:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
    /bin/bash
    04/07/2022 12:38 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4HEALTHCHECK
    &{["NONE"] "0s" "0s" "0s" '\x00'}
    04/07/2022 12:38 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=v0.2.0 MODELS_VER=v0.3.0 NVTAB_VER=v1.0.0 SYSTEMS_VER=v0.1.0 TF4REC_VER=v0.1.7 rm -rf /opt/conda/share/jupyter/lab/staging/node_modules/node-fetch
    04/07/2022 12:38 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=v0.2.0 MODELS_VER=v0.3.0 NVTAB_VER=v1.0.0 SYSTEMS_VER=v0.1.0 TF4REC_VER=v0.1.7 rm -rf /opt/conda/share/jupyter/lab/staging/node_modules/marked
    04/07/2022 12:38 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=v0.2.0 MODELS_VER=v0.3.0 NVTAB_VER=v1.0.0 SYSTEMS_VER=v0.1.0 TF4REC_VER=v0.1.7 rm -rf /repos
    04/07/2022 12:38 AM UTC
    sha256:68e235df0cab9e5938731ed6576efe12b7e17202f23fd88c9c00469fda525fd1RUN
    CORE_VER=v0.2.0 MODELS_VER=v0.3.0 NVTAB_VER=v1.0.0 SYSTEMS_VER=v0.1.0 TF4REC_VER=v0.1.7 git clone https://github.com/NVIDIA-Merlin/Merlin/ /Merlin
    04/07/2022 12:38 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PYTHONPATH=:/core:/systems:/nvtabular:/transformers4rec:/models
    04/07/2022 12:38 AM UTC
    sha256:e3a0e6ff6496a044be60dd3d8a0b8e3aa8a268b47f51e4e710c2a9e42e7bee22RUN
    CORE_VER=v0.2.0 MODELS_VER=v0.3.0 NVTAB_VER=v1.0.0 SYSTEMS_VER=v0.1.0 TF4REC_VER=v0.1.7 git clone https://github.com/NVIDIA-Merlin/Models.git /models/ &&
      cd /models/ &&
      git checkout ${MODELS_VER} &&
      pip install -e . --no-deps
    04/07/2022 12:38 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PYTHONPATH=:/core:/systems:/nvtabular:/transformers4rec
    04/07/2022 12:38 AM UTC
    sha256:bae17fb650970adc649a727a568f7508c9d919b84ddc91664c1bd0535671c4ceRUN
    CORE_VER=v0.2.0 MODELS_VER=v0.3.0 NVTAB_VER=v1.0.0 SYSTEMS_VER=v0.1.0 TF4REC_VER=v0.1.7 git clone https://github.com/NVIDIA-Merlin/Transformers4Rec.git /transformers4rec &&
      cd /transformers4rec/ &&
      git checkout ${TF4REC_VER} &&
      pip install -e . --no-deps
    04/07/2022 12:38 AM UTC
    ...