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: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
...