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

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

LayerLabelCreated
3a3b29e7f62e758cd3baf4c7d2088fced7c3e3bebbf029c1698e0317b34fd8e8RUN
|4
09/29/2023 2:49 PM UTC
c843d12dd1d3375e8d481f2752fc9e11de55cebf276904c1dd62b0de9762fdccRUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v23.09.00 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git ln -s /opt/tritonserver/backends/pytorch/* /usr/local/lib/
09/29/2023 2:49 PM UTC
7c8164cbfda0b54874e84b93831c4c48faa2f6ad257a7da864495ca06a9f5525RUN
HUGECTR_DEV_MODE=false HUGECTR_VER=v23.09.00 _CI_JOB_TOKEN= _HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git if [ "$HUGECTR_DEV_MODE" == "false" ]; then export HUGECTR_HOME=/usr/local/hugectr &&
  rm -rf ${HUGECTR_HOME}/lib/libgmock* ${HUGECTR_HOME}/lib/pkgconfig/gmock* ${HUGECTR_HOME}/include/gmock &&
  rm -rf ${HUGECTR_HOME}/lib/libgtest* ${HUGECTR_HOME}/lib/pkgconfig/gtest* ${HUGECTR_HOME}/include/gtest &&
  git clone --branch ${HUGECTR_VER} --depth 1 --recurse-submodules --shallow-submodules https://${_CI_JOB_TOKEN}${_HUGECTR_REPO} /hugectr &&
  pushd /hugectr/hps_torch/ &&
  pip --no-cache-dir install ninja &&
  TORCH_CUDA_ARCH_LIST="7.0 7.5 8.0 9.0" python setup.py install &&
  popd &&
  rm -rf /hugectr ; fi
09/29/2023 2:49 PM UTC
6eb986753ee6d290cd2ea2ef124c93cbbef62a842e1a2ac58f55380ee12833d4ARG
HUGECTR_VER=main
09/29/2023 2:47 PM UTC
624af7311db6ec8e8b60980b87ba35dad579c572a75e39c1099c55acf7f687d2ARG
_CI_JOB_TOKEN=
09/29/2023 2:47 PM UTC
dfb405e0706f5ba5b356be359653b453bb4d9b81c275fb712b08c870231e1001ARG
_HUGECTR_REPO=github.com/NVIDIA-Merlin/HugeCTR.git
09/29/2023 2:47 PM UTC
9394b504abf5429ac79ea921b7d7b017a1f4d0915c54fa65ae59942612afe8e2ARG
HUGECTR_DEV_MODE=false
09/29/2023 2:47 PM UTC
70d5c2f90f373c2adb2688f7330e9511feb4f3150934d37ac291f0e1ba53c786COPY
--chown=1000:1000dir:1a5ca6dfe8eada207b7ef8e793a3d89c8a45335e87a07bc147d9f535fd1a7426 in /usr/local/lib/python3.10/dist-packages/torch.egg-info/
09/29/2023 2:47 PM UTC
a05b5fbdb74299e2a0e00722b8f4c8d8c4769df464e23b65f7acd9893d2abfc9COPY
--chown=1000:1000dir:aa26a6b9ed90d8a30bd0209a98ead8506ba8005252f79b13314c46afe5dac6aa in /usr/local/lib/python3.10/dist-packages/numpy.dist-info/
09/29/2023 2:46 PM UTC
00cd470ac5fd33bed4285e5337d35369d02ad5e3e6feb2a35659c28992581a70COPY
--chown=1000:1000dir:7b4ffac94789f98e1ca57dc46d652ac800ab9018d6f0d8fe9197c71468fe53a7 in /usr/local/lib/python3.10/dist-packages/numba.dist-info/
09/29/2023 2:46 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.