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
49b90f76388411eca6adf4d5068dd446a1143171f7c7cf8014202a52b21e9204RUN
/bin/sh
12/31/2022 1:47 AM UTC
91ce51c1d326e045655f35429c7a26b185365e983700db2459cf623de6c6c015RUN
pip install --no-deps fastai fastcore fastprogress fastdownload
12/31/2022 1:47 AM UTC
1920aecacb35666913d57392bd5d8e60a30e1c87bb714084860ef094efadb881RUN
ln -s /opt/tritonserver/backends/pytorch/* /usr/local/lib/
12/31/2022 1:47 AM UTC
40fe83fd3e07f5a5f617b4f43585a16acc7a133aa6b6d82f7dde39d3913c5dcdCOPY
--chown=1000:1000dir:78caef98c724c087bc49df720cca19af83458677e6e9b4ba3be341187d03511b in /usr/local/lib/python3.8/dist-packages/torch
12/31/2022 1:47 AM UTC
68225f5c466156e76da2626971e4c553e4d15be43497d20859b9a8a91c7e163cCOPY
--chown=1000:1000dir:3b74f563a5b3bcb3371739465e8068e0ea2866d7cf613d60c13c3352d49fe32c in /usr/local/lib/python3.8/dist-packages/numpy
12/31/2022 1:47 AM UTC
1e871efe894dbeb6b8c09a4cc697972fcbf3fc7e4cbef732f2ab402ad9b53da6COPY
--chown=1000:1000dir:321b4265471393fd4d1240e6e3ba5037b21ab0161389bc133a9b4d3a7491a7fc in /usr/local/lib/python3.8/dist-packages/numba
12/31/2022 1:47 AM UTC
de57a549135dacb057f89a32953b862a1d3b82e082f11661b60d000662024846COPY
--chown=1000:1000dir:6e6aff9279812ba5ea0ecd4129555ab104f2d0c5cd4692269a2d63d14120a88a in backends/pytorch
12/31/2022 1:46 AM UTC
ff5b2641e3670c2131a6ded45377cca74859cc9c06b0922aa70b71df418c9eebRUN
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
9080e8eb7ec8a8103fa19b559489adec7d997b2b1689dcb098a53689aca5ece4RUN
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
90febf85086c8c29e85f3efd57346b07b27606c33dc590131de880fa25e11793ENTRYPOINT
/opt/nvidia/nvidia_entrypoint.sh
12/31/2022 12:57 AM UTC
...