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:2a723ce551da0493390ec86797ae26e9e61653b96bf981d8dfa80247b8e611c1RUN
pip install --no-cache-dir matplotlib
08/30/2023 3:03 AM UTC
sha256:b0935f2f040a005bdf354acd12bb1c97cdf8b794af124a9115c5d725964e79b2RUN
ln -s /opt/tritonserver/backends/pytorch/* /usr/local/lib/
08/30/2023 3:03 AM UTC
sha256:557e365fa9e2956bb2776950a01821620805e3c7c9b7bf145aa269bbf292a5e3COPY
--chown=1000:1000dir:1a5ca6dfe8eada207b7ef8e793a3d89c8a45335e87a07bc147d9f535fd1a7426 in /usr/local/lib/python3.10/dist-packages/torch.egg-info/
08/30/2023 3:03 AM UTC
sha256:0a197ab1b244e1fb309135589e09987430d0dbeb9405f84c9641148022eb1e10COPY
--chown=1000:1000dir:aa26a6b9ed90d8a30bd0209a98ead8506ba8005252f79b13314c46afe5dac6aa in /usr/local/lib/python3.10/dist-packages/numpy.dist-info/
08/30/2023 3:03 AM UTC
sha256:c27f31b94ff74b1bbc928bef18ea1a7e8c1c7c095f3eecb21f2e44baa2567c64COPY
--chown=1000:1000dir:7b4ffac94789f98e1ca57dc46d652ac800ab9018d6f0d8fe9197c71468fe53a7 in /usr/local/lib/python3.10/dist-packages/numba.dist-info/
08/30/2023 3:03 AM UTC
sha256:c17e7a59b72ead408e1a785f435e8199e2763f031d59b37b16ad837de5abecc8COPY
--chown=1000:1000dir:c0325a44c93e2b7ad6997bfa75ca0d31d2960146ea70aa2ab88053d224a73b2d in /usr/local/lib/python3.10/dist-packages/torch
08/30/2023 3:03 AM UTC
sha256:c15fb7a65e312af438bd8edcca614dd5a4adf3e691d4ae5db22e8aafa36377e8COPY
--chown=1000:1000dir:de02e7822881480fd557d72181023dc7d978956a2cd2476db41f4314a6d0b929 in /usr/local/lib/python3.10/dist-packages/numpy
08/30/2023 3:03 AM UTC
sha256:be73836a206c779b411097384dbec20545281b53749d0528cc533ab201fd5cfbCOPY
--chown=1000:1000dir:e22681d71bcec78ef3cf2f43e2ec9872962c78bbf1996c7aed2949e4536712fa in /usr/local/lib/python3.10/dist-packages/numba
08/30/2023 3:03 AM UTC
sha256:57ea180890b215adadc2f52038c5e3dddc56f744edd3c3aabad2a3d8b1bb96abCOPY
--chown=1000:1000dir:f9728e275bd8e3981ce2e4b61066d528c72248225697b2ac1e54aee46a12fce3 in backends/pytorch
08/30/2023 3:02 AM UTC
sha256:56f0689b94f7416b45e3be6c9b00e231ccb35242fd0df5dc9cf0dd5271b40db3RUN
pip install --no-cache-dir --no-deps torch torchmetrics pytorch-lightning lightning-utilities &&
  pip install --no-cache-dir --upgrade pip &&
  pip install sympy &&
  rm -rf /usr/local/lib/python${PYTHON_VERSION}/dist-packages/torch &&
  rm -rf /usr/local/lib/python${PYTHON_VERSION}/dist-packages/caffe2
08/30/2023 3:02 AM UTC
...