Skip to main content
NVIDIA
Merlin Training
Container
NVIDIA
Merlin Training

This container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with HugeCTR.

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
    /bin/bash
    03/04/2022 8:49 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4HEALTHCHECK
    &{["NONE"] "0s" "0s" "0s" '\x00'}
    03/04/2022 8:49 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    HUGECTR_DEV_MODE=false HUGECTR_VER=v3.4.1 HWLOC_VER=2.4.1 INSTALL_HDFS=false MODELS_VER=main NVTAB_VER=v0.11.0 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git rm -rf /usr/local/share/jupyter/lab/staging/node_modules/node-fetch
    03/04/2022 8:49 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    HUGECTR_DEV_MODE=false HUGECTR_VER=v3.4.1 HWLOC_VER=2.4.1 INSTALL_HDFS=false MODELS_VER=main NVTAB_VER=v0.11.0 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git rm -rf /usr/local/share/jupyter/lab/staging/node_modules/marked
    03/04/2022 8:49 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    HUGECTR_DEV_MODE=false HUGECTR_VER=v3.4.1 HWLOC_VER=2.4.1 INSTALL_HDFS=false MODELS_VER=main NVTAB_VER=v0.11.0 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git rm -rf /repos
    03/04/2022 8:49 AM UTC
    sha256:a1616776ff4b37594c4877a32cafd74466bf33972bb048a017ae004193e61eacRUN
    HUGECTR_DEV_MODE=false HUGECTR_VER=v3.4.1 HWLOC_VER=2.4.1 INSTALL_HDFS=false MODELS_VER=main NVTAB_VER=v0.11.0 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git pip install numba==0.53.1 numpy==1.22.2 --no-deps
    03/04/2022 8:49 AM UTC
    sha256:36268aa2e93bf83ea9933dfdb04a5fed35e648ba39240abeee4c61bacc636265RUN
    HUGECTR_DEV_MODE=false HUGECTR_VER=v3.4.1 HWLOC_VER=2.4.1 INSTALL_HDFS=false MODELS_VER=main NVTAB_VER=v0.11.0 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git rm /usr/local/cuda/lib64/stubs/libcuda.so.1
    03/04/2022 8:49 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PYTHONPATH=/hugectr/onnx_converter:/usr/local/hugectr/lib:/nvtabular:/core:
    03/04/2022 8:49 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    PYTHONPATH=/usr/local/hugectr/lib:/nvtabular:/core:
    03/04/2022 8:49 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENV
    LD_LIBRARY_PATH=/usr/local/hugectr/lib:/usr/java/jdk-16.0.2/lib/server
    03/04/2022 8:49 AM UTC
    ...