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
sha256:80a7df1370d9a2f3db396e0ab2376ceedd0b45fb1ab3d1dff6714be5e3382654RUN
pip install matplotlib
12/31/2022 1:47 AM UTC
sha256:1753efd846731ead3b14720894c3d5a6fdcd1f0541c0ac994f3aa7ae194f1c22RUN
pip install --no-deps fastai fastcore fastprogress fastdownload
12/31/2022 1:47 AM UTC
sha256:a11f7f1f576994c47bdab99f7c92564e4d601b82b19a2fed9e612671a6296d51RUN
ln -s /opt/tritonserver/backends/pytorch/* /usr/local/lib/
12/31/2022 1:47 AM UTC
sha256:d682c3f12ab242ae2e5adaa4b367988d93348ebdd00fbeb8e49f11fb545c1421COPY
--chown=1000:1000dir:78caef98c724c087bc49df720cca19af83458677e6e9b4ba3be341187d03511b in /usr/local/lib/python3.8/dist-packages/torch
12/31/2022 1:47 AM UTC
sha256:8758332a2f2e0eaee15165a685c38f9c41936a50e4f1e1b1546c36ae54d9a90fCOPY
--chown=1000:1000dir:3b74f563a5b3bcb3371739465e8068e0ea2866d7cf613d60c13c3352d49fe32c in /usr/local/lib/python3.8/dist-packages/numpy
12/31/2022 1:47 AM UTC
sha256:d3d31f076adabe64d1f1c81e48dc84315bb6ac657e47a94f7a4f08b39b289709COPY
--chown=1000:1000dir:321b4265471393fd4d1240e6e3ba5037b21ab0161389bc133a9b4d3a7491a7fc in /usr/local/lib/python3.8/dist-packages/numba
12/31/2022 1:47 AM UTC
sha256:aed1c7de1b9cb6e7dcbc366476b884abd77d15675561befdc9320fd2af2f1c79COPY
--chown=1000:1000dir:6e6aff9279812ba5ea0ecd4129555ab104f2d0c5cd4692269a2d63d14120a88a in backends/pytorch
12/31/2022 1:46 AM UTC
sha256:6f21c5dc422fbcd319c749d8531cf9c35fe95415bb412bd21baa4f45fffd8c19RUN
pip install --no-cache-dir --no-deps torch torchmetrics &&
  rm -rf /usr/local/lib/python3.8/dist-packages/torch &&
  rm -rf /usr/local/lib/python3.8/dist-packages/caffe2
12/31/2022 1:46 AM UTC
sha256:530b34a45ee7f9e95c275dd0742b6800fc78f17d32682aa47e964b280e94aa10RUN
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/*
12/31/2022 1:46 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
/opt/nvidia/nvidia_entrypoint.sh
12/31/2022 12:57 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.