Skip to main content
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: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
    ...

    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.