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:79e1c2016ee8bc240ab79851ccc17ee610a95227e0fa3262d3e96bc21ddc0bf7RUN
pip install matplotlib
07/27/2022 5:36 AM UTC
sha256:92a05b8d0085847b5bd0c51bff768e177d2924708e26d0227f05a1d93d8899d5RUN
pip install --no-deps fastai fastcore fastprogress fastdownload
07/27/2022 5:36 AM UTC
sha256:b2d63f7e75d03a620f77659bb74aad1f0d79750f3dbe8cca50bf2400e0c037b1COPY
--chown=1000:1000dir:71f0a7b2880dd910f2e1dfa60e2e44e8ddaf033115d8dba19cdb628052afcf84 in /usr/local/lib/python3.8/dist-packages/torch
07/27/2022 5:36 AM UTC
sha256:c1ccda5fa848afe10f4a12061d1540d244baf40c66cca9013da458162306fd88COPY
--chown=1000:1000dir:ab5ce33d27a23562798b929a6711c70f6e22f419cde4c5f0da6814042596bd21 in /usr/local/lib/python3.8/dist-packages/numpy
07/27/2022 5:36 AM UTC
sha256:a743065111414ef4830bd83623f90a33a6634082cd40582156db940bbffe40ebCOPY
--chown=1000:1000dir:254113775845405af1730557c3e681d3daa187225cd890a8567dd4341fb6b63d in /usr/local/lib/python3.8/dist-packages/numba
07/27/2022 5:36 AM UTC
sha256:4ecf6563a01df555fc39669fc003e4d109d85b00a82a583b9e9e80149e660df0COPY
--chown=1000:1000dir:b3170a86e45dc73f932bb1984479f1a02c341bf2f712e83957c90e9af17be4cc in backends/pytorch
07/27/2022 5:36 AM UTC
sha256:595bef43756edd08c51f7431ec7a50dab8551ae0e429476b663a7680202735c3RUN
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
sha256:b30b77dcb43be072420fed820d3aba7882f3029723ae998e19ff83802e6a263cRUN
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
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
/opt/nvidia/nvidia_entrypoint.sh
07/26/2022 8:46 AM UTC
sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
/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.