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:2b50190f5803bf4026d086b09749000b64d4d58df560500b3c99658337b5be53RUN
pip install --no-cache-dir matplotlib
03/09/2023 9:59 AM UTC
sha256:c87b741eccaa500787903e9f4b5c077aa21decb8feabba4ed62dbaaeb76cc6b1RUN
ln -s /opt/tritonserver/backends/pytorch/* /usr/local/lib/
03/09/2023 9:59 AM UTC
sha256:e981fe46be7cb772af96f2102ccecb849b7592969fe21e0e25f7c89b243a0244COPY
--chown=1000:1000dir:438be91ff4a7f2eb6969bcfa9102e396198048d4df5c0589c82a23652c33c3ad in /usr/local/lib/python3.8/dist-packages/torch.egg-info/
03/09/2023 9:59 AM UTC
sha256:df82307df16968ae0cc6f89df4971b0d9171aaa3cf9023ed1890ea3e2384c7d3COPY
--chown=1000:1000dir:b2533420e33d8337afde9d69ec6d5ff238705ae597848613d2e87b98aa759ecf in /usr/local/lib/python3.8/dist-packages/numpy.dist-info/
03/09/2023 9:59 AM UTC
sha256:5096c824a7fc0b63bee5bdf58d2072d44c3f604231670ab374057b4ce6a90d93COPY
--chown=1000:1000dir:c3e274754b9ed44f4837970f248064bce476b1a8e454ea7008bf3c28141900df in /usr/local/lib/python3.8/dist-packages/numba.dist-info/
03/09/2023 9:59 AM UTC
sha256:e589798886807a985191bab00a78bb6c166447a9ab65ffbb61002fd0a36a416bCOPY
--chown=1000:1000dir:0b4a2fe3316b0af7a990d225be3c362e6ae530729d8087392236a34d44f12e83 in /usr/local/lib/python3.8/dist-packages/torch
03/09/2023 9:59 AM UTC
sha256:d7f02fe7b783c9485694b91e87f4c2c3d4399bad4f6a0de4b08d98b10b29201eCOPY
--chown=1000:1000dir:35d5a1faa50a089662e762b1f0b5bf2a3838e42c7d9f7942de2a3c5cc7e52041 in /usr/local/lib/python3.8/dist-packages/numpy
03/09/2023 9:58 AM UTC
sha256:b23d4c149b0f716b17c90c214150a383911967a49633720b5f475167013d035cCOPY
--chown=1000:1000dir:b794a05047dcc4f20e757b8295ff0e7a362ce60eeb726b9af9a1d45604e081c0 in /usr/local/lib/python3.8/dist-packages/numba
03/09/2023 9:58 AM UTC
sha256:d4b43927463ffc202c522d1644a09903cdc65f9c824f1977bfa48845047fb853COPY
--chown=1000:1000dir:f5e4d3a54d12a65ddfeefa2b71bd2a6418bf7eb551bc4277195322230e62b76b in backends/pytorch
03/09/2023 9:58 AM UTC
sha256:39084b743e5c3797630bc9e0023300f2ee2d1cc2eacdc3c78545e11ca99617a2RUN
pip install --no-cache-dir --no-deps torch torchmetrics &&
  pip install --no-cache-dir --upgrade pip &&
  rm -rf /usr/local/lib/python3.8/dist-packages/torch &&
  rm -rf /usr/local/lib/python3.8/dist-packages/caffe2
03/09/2023 9:58 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.