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
NVIDIA
Merlin TensorFlow
Container
NVIDIA
NVIDIA
Merlin TensorFlow

The Merlin TensorFlow container allows users to do preprocessing and feature engineering with NVTabular, and then train a deep-learning based recommender system model with TensorFlow, and serve the trained model on Triton Inference Server.

LayerLabelCreated
sha256:8feec0e2f21fd8365de47af1abd69a70e97962233e9e357cee41f991b1f550c9RUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v4.1 INSTALL_DISTRIBUTED_EMBEDDINGS=true TFDE_VER=v0.2 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git if [ "$INSTALL_DISTRIBUTED_EMBEDDINGS" == "true" ]; then git clone https://github.com/NVIDIA-Merlin/distributed-embeddings.git /distributed_embeddings/ &&
  cd /distributed_embeddings &&
  git checkout ${TFDE_VER} &&
  git submodule update --init --recursive &&
  make pip_pkg &&
  pip install artifacts/*.whl &&
  make clean; fi
10/27/2022 12:56 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
TFDE_VER=v0.2
10/27/2022 12:54 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
INSTALL_DISTRIBUTED_EMBEDDINGS=true
10/27/2022 12:54 PM UTC
sha256:74596a75b5a8b356c1800eacee9d28f4d47280e61ef760cf05080c05309db651RUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v4.1 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git if [ "$HUGECTR_DEV_MODE" == "false" ]; then git clone --branch ${HUGECTR_VER} --depth 1 https://${_CI_JOB_TOKEN}${_HUGECTR_REPO} /hugectr &&
  pushd /hugectr &&
  pip install ninja &&
  git submodule update --init --recursive &&
  cd sparse_operation_kit &&
  python setup.py install &&
  cd ../hierarchical_parameter_server &&
  python setup.py install &&
  popd; fi
10/27/2022 12:54 PM UTC
sha256:a61a7183cf15fe7f9c7e06889800f7f87f38fdf1640c3d5826fedd2c78d747bbRUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v4.1 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git ln -s /usr/lib/x86_64-linux-gnu/libibverbs.so.1 /usr/lib/x86_64-linux-gnu/libibverbs.so
10/27/2022 12:43 PM UTC
sha256:9a03484ce70821ce2d15cd699e7ab5641680d9a39e008b9bcc0c188243a78b00RUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v4.1 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git mkdir -p /usr/local/nvidia/lib64 &&
  ln -s /usr/local/cuda/lib64/libcusolver.so /usr/local/nvidia/lib64/libcusolver.so.10
10/27/2022 12:43 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
HUGECTR_VER=main
10/27/2022 12:43 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
_CI_JOB_TOKEN=
10/27/2022 12:43 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
_HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git
10/27/2022 12:43 PM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ARG
HUGECTR_DEV_MODE=false
10/27/2022 12:43 PM UTC
...