ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIARiva Skills Quick Start
Helm ChartsNVIDIAGPU OperatorNVIDIANVIDIA NIM Operator
  • 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
    		03/04/2022 8:07 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4HEALTHCHECK
    &{["NONE"] "0s" "0s" "0s" '\x00'}
    		03/04/2022 8:07 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=main DASK_VER=2021.11.2 MODELS_VER=main NVTAB_VER=v0.11.0 TF4REC_VER=v0.1.6 rm -rf /opt/conda/share/jupyter/lab/staging/node_modules/node-fetch
    		03/04/2022 8:07 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=main DASK_VER=2021.11.2 MODELS_VER=main NVTAB_VER=v0.11.0 TF4REC_VER=v0.1.6 rm -rf /opt/conda/share/jupyter/lab/staging/node_modules/marked
    		03/04/2022 8:07 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    CORE_VER=main DASK_VER=2021.11.2 MODELS_VER=main NVTAB_VER=v0.11.0 TF4REC_VER=v0.1.6 rm -rf /repos
    		03/04/2022 8:07 AM UTC
    sha256:6451bd6a9d2c9ec16563ee5ac224b078da97a43d69b94562e4fb030fda97be7dRUN
    CORE_VER=main DASK_VER=2021.11.2 MODELS_VER=main NVTAB_VER=v0.11.0 TF4REC_VER=v0.1.6 pip install numba==0.53.1 numpy==1.22.2 --no-deps
    		03/04/2022 8:07 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PYTHONPATH=/models:/transformers4rec:/nvtabular:/core:
    		03/04/2022 8:07 AM UTC
    sha256:4e09d1c3699fe01c9bc1b5acfb857e43cb5952a5bf585ea4955701a53e1e94aaRUN
    CORE_VER=main DASK_VER=2021.11.2 MODELS_VER=main NVTAB_VER=v0.11.0 TF4REC_VER=v0.1.6 git clone https://github.com/NVIDIA-Merlin/Models.git /models/ &&
  cd /models/ &&
  git checkout ${MODELS_VER} &&
  pip install -e . --no-deps
    		03/04/2022 8:07 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PYTHONPATH=/transformers4rec:/nvtabular:/core:
    		03/04/2022 8:06 AM UTC
    sha256:f64202b4a7574e9112bdb00dc71f9ea8fdc36cceb5b0165b9f20e4077a78ab31RUN
    CORE_VER=main DASK_VER=2021.11.2 MODELS_VER=main NVTAB_VER=v0.11.0 TF4REC_VER=v0.1.6 git clone https://github.com/NVIDIA-Merlin/Transformers4Rec.git /transformers4rec &&
  cd /transformers4rec/ &&
  git checkout ${TF4REC_VER} &&
  pip install -e . --no-deps
    		03/04/2022 8:06 AM UTC
    ...