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
f6bf58f772d39449ee359aa2694d89f7acf7e4077fd4c573a21344cfd629a56fRUN
/bin/sh
09/30/2022 5:34 AM UTC
befe7d1d1c72224760339f78a3fa539612b0e9ccbef4fc84b784a42364604882RUN
pip install --no-deps fastai fastcore fastprogress fastdownload
09/30/2022 5:34 AM UTC
cc64fdedf8ace9663237840057941d483fe2dda97b4b721592174ed9dd3261f2COPY
--chown=1000:1000dir:6f07987f8cb6c3fd3725beed0ea1a5861d4101e15c8f20c12e58d706191e86d8 in /usr/local/lib/python3.8/dist-packages/torch
09/30/2022 5:34 AM UTC
df79ff4a23618e9f90c75f7cdc668d7c15405eaf1a06b3d87c26c44d10e94013COPY
--chown=1000:1000dir:76386bde9f857a3d3b759e6831ae01ff0f76142feea94e09ab3e483616190c0a in /usr/local/lib/python3.8/dist-packages/numpy
09/30/2022 5:34 AM UTC
18709b5a81f05b59b51fb347a23883b00a0161c86590a12866547abd3d93244cCOPY
--chown=1000:1000dir:226bfeb0fa603039a20a2b22903990a7a8daf610d08ef18f88505dce3e0d9903 in /usr/local/lib/python3.8/dist-packages/numba
09/30/2022 5:34 AM UTC
238f07b144749a1d6780f7b9a958e783f1a558ea08a2d6103d82617d3bede699COPY
--chown=1000:1000dir:03e8d035af9fb9a9f9ea3cf98748c267ccca444c9d7bf3886ed3aca92dccbdb1 in backends/pytorch
09/30/2022 5:34 AM UTC
cfbb8722a76b64145771b41c61303f8d707103cab1537309776aeb9a59211793RUN
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
09/30/2022 5:33 AM UTC
178c4737beb352905a3f139b5b63645a79708cf779e42a242d5ccc55e760d7ccRUN
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/*
09/30/2022 5:33 AM UTC
e3c12bdd47786b208d2622312e58dd66108b2a673ffaccbbd6fbceb6799471d0ENTRYPOINT
/opt/nvidia/nvidia_entrypoint.sh
09/28/2022 6:40 AM UTC
b900ea97622e36248bacf8fb6658abc178aa91186b9392094e8a274ea8c31a47CMD
/bin/bash
09/28/2022 6:40 AM UTC
...