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