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
451dc9fab86fb5d7f9c07d86a8dda8d1781e96a4d372acf5df74baf2b0fac5abRUN
/bin/sh
07/27/2022 5:36 AM UTC
7e651ad7f452e8ead3e886645e08d17e5a3f0f91bf229dc7ee0abaee40cfce62RUN
pip install --no-deps fastai fastcore fastprogress fastdownload
07/27/2022 5:36 AM UTC
eb76a4f4fc31c466688778e557dfb60552640d90b39b93107f0492e339bc9c21COPY
--chown=1000:1000dir:71f0a7b2880dd910f2e1dfa60e2e44e8ddaf033115d8dba19cdb628052afcf84 in /usr/local/lib/python3.8/dist-packages/torch
07/27/2022 5:36 AM UTC
0344f9f750f14a02a56f9660e42111ee1f4566a9390a7c59a459b4efb0a74a4dCOPY
--chown=1000:1000dir:ab5ce33d27a23562798b929a6711c70f6e22f419cde4c5f0da6814042596bd21 in /usr/local/lib/python3.8/dist-packages/numpy
07/27/2022 5:36 AM UTC
a3b9d5271d0a357eed3b8aef3f88c7480a2b8b3487a7245726ca41246918f190COPY
--chown=1000:1000dir:254113775845405af1730557c3e681d3daa187225cd890a8567dd4341fb6b63d in /usr/local/lib/python3.8/dist-packages/numba
07/27/2022 5:36 AM UTC
500d39fbe6087a820c822338f899a3838b4b1f32bca475da568db98f7e2b4098COPY
--chown=1000:1000dir:b3170a86e45dc73f932bb1984479f1a02c341bf2f712e83957c90e9af17be4cc in backends/pytorch
07/27/2022 5:36 AM UTC
d83ea8befa3e26dad5632e76c9e7d746844f81608d201c6574513fbca373a9ceRUN
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
07/27/2022 5:36 AM UTC
bb39a024a39960affbaf662cec020f4377b09b04fb71de6c45b733483ba9a39bRUN
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/*
07/27/2022 5:35 AM UTC
6fb892ce229bdf0397460377a645cbf1dab478275ab55538110a7d5ef2bbc091ENTRYPOINT
/opt/nvidia/nvidia_entrypoint.sh
07/26/2022 8:46 AM UTC
382f1093a5e9a5e3c990224495789580fbae8b27d70c61e7555e770c22b126c8CMD
/bin/bash
07/26/2022 8:46 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.