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
    Merlin PyTorch
    Container
    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:2ce3fa9acf565bd04d79a8b713e25c2ac7e0beb4e3b18107bf125bb7af88114aRUN
    pip install --no-cache-dir matplotlib
    		06/23/2023 8:44 AM UTC
    sha256:a30124d883236c970f85b7a8ace486e06bc83757c3b791762849c6c20a5fb81dRUN
    ln -s /opt/tritonserver/backends/pytorch/* /usr/local/lib/
    		06/23/2023 8:44 AM UTC
    sha256:7e5907edce47b5b08931625c54bd9c0bbdee6b65d127be8024287f21c94e7405COPY
    --chown=1000:1000dir:5cf68f527375285fdf68201eabe6366bf3bbc33aa11df8b42ae00ab9af8f8208 in /usr/local/lib/python3.8/dist-packages/torch.egg-info/
    		06/23/2023 8:44 AM UTC
    sha256:b10097f6372b113eb5ae1e5f34b41ae4d7af6fb135a815586aae0d524e8c4f1cCOPY
    --chown=1000:1000dir:b2533420e33d8337afde9d69ec6d5ff238705ae597848613d2e87b98aa759ecf in /usr/local/lib/python3.8/dist-packages/numpy.dist-info/
    		06/23/2023 8:44 AM UTC
    sha256:542f0cd87070bd9081f2d4a5a16373c55dc7f6fe9728ecbfa88e0ca101d6f289COPY
    --chown=1000:1000dir:0c46506878b5c04715c97591280ce9a5cb69839344f1f831aa8b937e0a88703c in /usr/local/lib/python3.8/dist-packages/numba.dist-info/
    		06/23/2023 8:44 AM UTC
    sha256:6c415e8b64ed1bbbf3d38998259f6e29a01e6bb5f440dc95a6adacc59b553ad3COPY
    --chown=1000:1000dir:618204a1b808a866fc133d2a635c8bc0ddcaca93e5103c5697324a4c245ef2ad in /usr/local/lib/python3.8/dist-packages/torch
    		06/23/2023 8:44 AM UTC
    sha256:42622108bc59c9a8f6e18ec23717c22eb22e1ad9f9ab4328d5e43f9e73746c87COPY
    --chown=1000:1000dir:43abf050df1f2effae3e181f78a715fb2557ed6f02604af8031dc368595bdc80 in /usr/local/lib/python3.8/dist-packages/numpy
    		06/23/2023 8:44 AM UTC
    sha256:728a725359cee0469cb8eb7344fc185d352ab17a6f1d864f06c7a07b08c2def9COPY
    --chown=1000:1000dir:de2b0fd7148a8d2651b4074bc586826623910ecb2f8f24bea29bfc8667d3267a in /usr/local/lib/python3.8/dist-packages/numba
    		06/23/2023 8:44 AM UTC
    sha256:f9afe432dbe786aedaa799a17e74677d1283db47bbba3bf8523e0712c8127f74COPY
    --chown=1000:1000dir:6581328c6900e6f21ce722c57d431bef162fe28048b315a276c8c4ffc8a8c9a6 in backends/pytorch
    		06/23/2023 8:44 AM UTC
    sha256:b197676f1ee9a6052b39e8a3fb96eb1d83ce3c4aaaf752a5ce1606bc4fcc8baeRUN
    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/python3.8/dist-packages/torch &&
  rm -rf /usr/local/lib/python3.8/dist-packages/caffe2
    		06/23/2023 8:44 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.