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/12/2022 9:03 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
    /bin/bash
    05/12/2022 9:03 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4HEALTHCHECK
    &{["NONE"] "0s" "0s" "0s" '\x00'}
    05/12/2022 9:03 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=v0.3.0 MODELS_VER=v0.4.0 NVTAB_VER=v1.1.1 SYSTEMS_VER=v0.2.0 TF4REC_VER=v0.1.8 rm -rf /opt/conda/share/jupyter/lab/staging/node_modules/node-fetch
    05/12/2022 9:03 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=v0.3.0 MODELS_VER=v0.4.0 NVTAB_VER=v1.1.1 SYSTEMS_VER=v0.2.0 TF4REC_VER=v0.1.8 rm -rf /opt/conda/share/jupyter/lab/staging/node_modules/marked
    05/12/2022 9:03 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=v0.3.0 MODELS_VER=v0.4.0 NVTAB_VER=v1.1.1 SYSTEMS_VER=v0.2.0 TF4REC_VER=v0.1.8 rm -rf /repos
    05/12/2022 9:03 AM UTC
    sha256:cc8f8008cbbe7114310f5ae5e8a5e8b790d5b0025ba4019369c14d6a64175bc0RUN
    CORE_VER=v0.3.0 MODELS_VER=v0.4.0 NVTAB_VER=v1.1.1 SYSTEMS_VER=v0.2.0 TF4REC_VER=v0.1.8 git clone https://github.com/NVIDIA-Merlin/Merlin/ /Merlin
    05/12/2022 9:03 AM UTC
    sha256:9559eb8aca5783e0d962ac58b60cf2f5cd33426356ee5708d0a4903058c1f94cRUN
    CORE_VER=v0.3.0 MODELS_VER=v0.4.0 NVTAB_VER=v1.1.1 SYSTEMS_VER=v0.2.0 TF4REC_VER=v0.1.8 git clone https://github.com/NVIDIA-Merlin/Models.git /models/ &&
      cd /models/ &&
      git checkout ${MODELS_VER} &&
      pip install . --no-deps
    05/12/2022 9:03 AM UTC
    sha256:74287890e8863a8b07ba6e75b735156e09dfaeca90ce76c3f049d0b31ac71815RUN
    CORE_VER=v0.3.0 MODELS_VER=v0.4.0 NVTAB_VER=v1.1.1 SYSTEMS_VER=v0.2.0 TF4REC_VER=v0.1.8 git clone https://github.com/NVIDIA-Merlin/Transformers4Rec.git /transformers4rec &&
      cd /transformers4rec/ &&
      git checkout ${TF4REC_VER} &&
      pip install . --no-deps
    05/12/2022 9:02 AM UTC
    sha256:c948935e1d23df53a8eec9a22e3a0570ba12db98b1173cf759e1dc62edc4f6b3RUN
    CORE_VER=v0.3.0 MODELS_VER=v0.4.0 NVTAB_VER=v1.1.1 SYSTEMS_VER=v0.2.0 TF4REC_VER=v0.1.8 git clone https://github.com/NVIDIA-Merlin/NVTabular.git /nvtabular/ &&
      cd /nvtabular/ &&
      git checkout ${NVTAB_VER} &&
      pip install . --no-deps
    05/12/2022 9:02 AM UTC
    ...