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: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
    ...