Skip to main content
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 Tensorflow Training
    Container
    NVIDIA
    Merlin Tensorflow Training

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

  • LayerLabelCreated
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) CMD ["/bin/bash"]
    		11/04/2021 5:22 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) ENTRYPOINT []
    		11/04/2021 5:22 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) HEALTHCHECK &{["NONE"] "0s" "0s" "0s" '\x00'}
    		11/04/2021 5:22 AM UTC
    sha256:1cf44ad041d95812dff2107ab8c905609f03bfd5c38a6b078b2b71d784cb783dRUN
    HUGECTR_VER=v3.2.1 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c echo $(du -h --max-depth=1 /)
    		11/04/2021 5:22 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4RUN
    /bin/bash -c #(nop) SHELL [/bin/bash -c]
    		11/04/2021 5:22 AM UTC
    sha256:87400c2e4a1fa04c3bab3308efbb45ca0ef15c04e47975dce170e438756dcb30RUN
    HUGECTR_VER=v3.2.1 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c rm -rf /usr/local/share/jupyter/lab/staging/node_modules/fast-json-patch
    		11/04/2021 5:22 AM UTC
    sha256:d83a006981496c09b1ff946ff03195554f97c4443071ce9d5798e56f03ae9400RUN
    HUGECTR_VER=v3.2.1 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c pip install numba numpy --upgrade
    		11/04/2021 5:22 AM UTC
    sha256:3ce77a946b2d558bdac99bed84be353a159dc083b61e40ae1d9ec289f3fa1ce1RUN
    HUGECTR_VER=v3.2.1 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c pip install pybind11
    		11/04/2021 5:21 AM UTC
    sha256:eed1f71192afd694d412543eac0b8d8626ea228c57e063c93ce1da66da211094RUN
    HUGECTR_VER=v3.2.1 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c git clone https://github.com/NVIDIA-Merlin/HugeCTR.git build-env &&
  pushd build-env &&
  if [ "$RELEASE" == "true" ] &&
  [ ${HUGECTR_VER} != "vnightly" ] ; then git fetch --all --tags &&
  git checkout tags/${HUGECTR_VER}; else echo ${HUGECTR_VER} &&
  git checkout master; fi &&
  cd sparse_operation_kit &&
  bash ./install.sh --SM=$SM --USE_NVTX=$USE_NVTX &&
  popd &&
  rm -rf build-env &&
  rm -rf /var/tmp/HugeCTR
    		11/04/2021 5:21 AM UTC
    sha256:6c105208164d457ab00ee90571b164f402be5916efb66087c2f4d741418f66a2RUN
    HUGECTR_VER=v3.2.1 RELEASE=true SM=60;61;70;75;80 USE_NVTX=OFF /bin/bash -c ln -s /usr/lib/x86_64-linux-gnu/libibverbs.so.1 /usr/lib/x86_64-linux-gnu/libibverbs.so
    		11/04/2021 5:19 AM UTC
    ...