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
d6b6ddd2a9ab09ed02e1289be8f70ae4a0809df1821cf2ffc2173f0ccf0a4c07RUN
|6
10/27/2022 12:56 PM UTC
4de8bdd885503d4a7c3495e3f466bc18295ad7882486adf7d23c15abe107a70bARG
TFDE_VER=v0.2
10/27/2022 12:54 PM UTC
252caefa1e2ac678ed4b75591bfc01232d77760039d874a0f55a4757c1463805ARG
INSTALL_DISTRIBUTED_EMBEDDINGS=true
10/27/2022 12:54 PM UTC
609b21cce2c5a7722ac886a46605d12d2e14902273a383c47a039581d9487568RUN
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
a270ee83a45f760b2325d3d2b70a9e909b20c996266d9b0dbf6e500b59627604RUN
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
7bbc87338bc84ebba4c50bca04f51af54e544fe96c81bb09a7c6c6b2ae3e91c2RUN
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
ef60e95f4d47b4d154c0ba341b99ad7ab7fd3ea2a6ccb9898c0de35299366289ARG
HUGECTR_VER=main
10/27/2022 12:43 PM UTC
d6fba041bd13492765cb71f422564af2c880b028a7fb56e8eb4690a5f014f04fARG
_CI_JOB_TOKEN=
10/27/2022 12:43 PM UTC
1cb6f7c964703895ade52dae8e61987d7ac871558164837a7ea85e9049009ab2ARG
_HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git
10/27/2022 12:43 PM UTC
e38208f779880e2a811668e255066d3c5c941d6420d0006e8e96da1a208bdf87ARG
HUGECTR_DEV_MODE=false
10/27/2022 12:43 PM UTC
...

NVIDIA uses cookies to improve your experience on our web site. We and our third-party partners also use cookies and other tools to collect and record information you provide as well as information about your interactions with our websites for performance improvement, analytics, and to assist in marketing efforts. By clicking "Accept All", you consent to our use of cookies and other tools as described in our Cookie Policy. You can manage your cookie settings by clicking on "Manage Settings." By continuing to use this site or by clicking one of the buttons below, you agree to our Terms of Service (which contains important waivers). Please see our Privacy Policy for more information on our privacy practices.