ContainersNVIDIADeepSeek-R1NVIDIALlama-3.1-Nemotron-70B-InstructNVIDIAPyTorch
CollectionsNVIDIAOmniverse Kit (FB)NVIDIADeepStream SDKNVIDIAOmniverse Kit App Streaming
ModelsNVIDIAStyleGAN3 pretrained modelsNVIDIAPeopleNetNVIDIATrafficCamNet
ResourcesNVIDIAKit SDK - Windows (PB25h1)NVIDIAKit SDK - Linux (PB25h1)NVIDIARiva 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
5fae5a85ac638f013d5e526ff6eefde2a689d5d201188684a82422e2689d33dcRUN
|4
08/01/2022 1:09 PM UTC
7c679483601f41a84887a57c07550465a6e624d6326d9407a57797a8adb56214ARG
INSTALL_DISTRIBUTED_EMBEDDINGS=true
08/01/2022 1:08 PM UTC
e5a5aeb3765de4f7526cfe5c75b39a89eb9ff77f58438ae8130652d6c5180ae6RUN
HUGECTR_DEV_MODE=false _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git if [ "$HUGECTR_DEV_MODE" == "false" ]; then git clone https://${_CI_JOB_TOKEN}${_HUGECTR_REPO} /hugectr &&
  pushd /hugectr &&
  git checkout ${HUGECTR_VER} &&
  cd sparse_operation_kit &&
  python setup.py install &&
  popd; fi
08/01/2022 1:08 PM UTC
23e289346bcbb381af9dd9cb1ef7dc9db6a5fdcf73aade139d5a12f6ff747232RUN
HUGECTR_DEV_MODE=false _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
08/01/2022 12:58 PM UTC
f147b75d6e970eb4f114ff3b42c91f0a60d90c68837281fa51f8d2d99d8b318dRUN
HUGECTR_DEV_MODE=false _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
08/01/2022 12:58 PM UTC
a5cf7295ca5e9da3cdc1aef411b02b18fdeeb43c71d59919fe7d9c8786050920ARG
_CI_JOB_TOKEN=
08/01/2022 12:58 PM UTC
00b8d001b78390628f1605892c973e36c8abbe634371f2680ce329e576d14180ARG
_HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git
08/01/2022 12:58 PM UTC
49cbefb345067cfc2fb4c84dbe2a7e6f2937a49384bb2a483f986e729a01fa82ARG
HUGECTR_DEV_MODE=false
08/01/2022 12:58 PM UTC
c014fb323db649f534f4b6a3e8f4df4abb3e0abffa170c55b5c6bd846d8b9b51ENV
LD_LIBRARY_PATH=/usr/local/hugectr/lib:/usr/local/cuda/compat/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64:/usr/local/lib:/repos/dist/lib:/opt/tritonserver/lib LIBRARY_PATH=/usr/local/hugectr/lib:/usr/local/cuda/lib64/stubs: SOK_COMPILE_UNIT_TEST=ON
08/01/2022 12:58 PM UTC
c4ed7b4deefdec5ab6fb271b79490b2583aa15fe8d58a4cd88ab176198f77d72RUN
wget http://www.cmake.org/files/v3.21/cmake-3.21.1.tar.gz &&
  tar xf cmake-3.21.1.tar.gz &&
  cd cmake-3.21.1 &&
  ./configure &&
  make &&
  make install
08/01/2022 12:58 PM UTC
...