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:92abc660473759915c6991fa98b9326f4a73ac811200593c593616dbb6443605RUN
pip install --no-cache-dir matplotlib
05/31/2023 7:25 PM UTC
sha256:80a9829ad0c5bc4b6908b6a1b9c7e58db3349b5be6f5a264ca7e66bd1aeec152RUN
ln -s /opt/tritonserver/backends/pytorch/* /usr/local/lib/
05/31/2023 7:24 PM UTC
sha256:da9dc273f76f047d719ee75032d814839be32870e9784a61341ec3d6f1c7e037COPY
--chown=1000:1000dir:41e5b2af01ec677f7794601e0ce75f8f9f556e54e93130327a5417fb7ff58ec1 in /usr/local/lib/python3.8/dist-packages/torch.egg-info/
05/31/2023 7:24 PM UTC
sha256:820fc3f4be34b30ef6128bca35b378adad9724c0bada71ebfab0c6062fbfc041COPY
--chown=1000:1000dir:b2533420e33d8337afde9d69ec6d5ff238705ae597848613d2e87b98aa759ecf in /usr/local/lib/python3.8/dist-packages/numpy.dist-info/
05/31/2023 7:24 PM UTC
sha256:1d1a23a9d3b4d93d1b61636a66a69b3700d1d0804dc001ebe3bd0ad5bb5308c5COPY
--chown=1000:1000dir:4169deef179af3fe4bada7e86e5238911cef9455eb4ddac2865576eaf863bc49 in /usr/local/lib/python3.8/dist-packages/numba.dist-info/
05/31/2023 7:24 PM UTC
sha256:87fbb4f4eec6a1a65288fc84c6132e0203d61c0133d0b34de7cc78537d81cbb8COPY
--chown=1000:1000dir:211e5315f6a46f8c430a669a8d36e3ec0135dcb6b4454da5cf6be67850c65da3 in /usr/local/lib/python3.8/dist-packages/torch
05/31/2023 7:24 PM UTC
sha256:1a850a58036f38f41c160efe4c8160ac3809d991ff11634b305fc81d60293094COPY
--chown=1000:1000dir:bf1c32fd3ef6d490b2efc47ee8d2038d8b49346d4db39a67df6c3903d5ccfcd8 in /usr/local/lib/python3.8/dist-packages/numpy
05/31/2023 7:24 PM UTC
sha256:4f7a4c7963091600a3380d7635116a087c904d4011294ea73e23d492e407a6b7COPY
--chown=1000:1000dir:f3c2c20c90fd3d637fba1bcf5e633b1e331646c3bfd6cbcd8c659e0d2ab43ee8 in /usr/local/lib/python3.8/dist-packages/numba
05/31/2023 7:24 PM UTC
sha256:97d4d0337426b22d8bbdba73d8b7c2fa3ff5716e8fc96822d36767694ef96e3eCOPY
--chown=1000:1000dir:44b1bcfb484d7a5b0dd769e0530d6e1a3c53f3fd8830e3467ef8ad75b613abe6 in backends/pytorch
05/31/2023 7:24 PM UTC
sha256:65fcc2556ef374f33e223b59e0341e66aa017e90b1f864fb65653e7483fedb56RUN
pip install --no-cache-dir --no-deps torch torchmetrics &&
  pip install --no-cache-dir --upgrade pip &&
  pip install sympy &&
  rm -rf /usr/local/lib/python3.8/dist-packages/torch &&
  rm -rf /usr/local/lib/python3.8/dist-packages/caffe2
05/31/2023 7:24 PM 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.