NVIDIA
NVIDIA
Merlin PyTorch Training
Container
NVIDIA
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
...