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
20aa07f6a0da52add6a1dc18f5b707a2520ce1ec2d5073ddad864bf8f478d355RUN
/bin/sh
10/27/2022 5:36 AM UTC
c7edd79afd36ee50926d63696cdb17392640b199ba5e98a60dee015830bce135RUN
pip install --no-deps fastai fastcore fastprogress fastdownload
10/27/2022 5:36 AM UTC
f456d550dca1b71499b41e6311f8000b59f9f5983ff6287ec0e04e529da5cf54COPY
--chown=1000:1000dir:6f07987f8cb6c3fd3725beed0ea1a5861d4101e15c8f20c12e58d706191e86d8 in /usr/local/lib/python3.8/dist-packages/torch
10/27/2022 5:36 AM UTC
382f3e53a841d752a3060c8018131710f783e52c6108b94b876401e88c779200COPY
--chown=1000:1000dir:76386bde9f857a3d3b759e6831ae01ff0f76142feea94e09ab3e483616190c0a in /usr/local/lib/python3.8/dist-packages/numpy
10/27/2022 5:36 AM UTC
ad0ff2c00974ee0ebdc251b129484f8455bed0ca6432c4156232dccd965d8b20COPY
--chown=1000:1000dir:226bfeb0fa603039a20a2b22903990a7a8daf610d08ef18f88505dce3e0d9903 in /usr/local/lib/python3.8/dist-packages/numba
10/27/2022 5:36 AM UTC
a3bded403b5d42fcf02691c9f1644269dc1bad3f852616e28e09537b5efb33fdCOPY
--chown=1000:1000dir:03e8d035af9fb9a9f9ea3cf98748c267ccca444c9d7bf3886ed3aca92dccbdb1 in backends/pytorch
10/27/2022 5:36 AM UTC
96e5a4c0ff5731943fd9781c4ca4484c946663450e212fed8ebb9d54db0a4aafRUN
pip install --no-cache-dir --no-deps torch torchmetrics &&
  rm -r /usr/local/lib/python3.8/dist-packages/torch &&
  rm -r /usr/local/lib/python3.8/dist-packages/caffe2
10/27/2022 5:35 AM UTC
b69487eac6640ee6841477c0d2432f1ad02f77749e694df99ecb9b3597b68b1fRUN
apt update -y --fix-missing &&
  apt install -y --no-install-recommends libatlas-base-dev libmkl-dev &&
  apt autoremove -y &&
  apt clean &&
  rm -rf /var/lib/apt/lists/*
10/27/2022 5:35 AM UTC
ca17102be5773ca89703e600751c408d228f9a0e2dc23244ab61e54f3875034aENTRYPOINT
/opt/nvidia/nvidia_entrypoint.sh
10/26/2022 6:42 AM UTC
1ff117ce76be9fb5f2fc7ca384993bc51d2aa3d9f06d009cb6f380bd42179a41CMD
/bin/bash
10/26/2022 6:42 AM 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.