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:80a7df1370d9a2f3db396e0ab2376ceedd0b45fb1ab3d1dff6714be5e3382654RUN
    pip install matplotlib
    12/31/2022 1:47 AM UTC
    sha256:1753efd846731ead3b14720894c3d5a6fdcd1f0541c0ac994f3aa7ae194f1c22RUN
    pip install --no-deps fastai fastcore fastprogress fastdownload
    12/31/2022 1:47 AM UTC
    sha256:a11f7f1f576994c47bdab99f7c92564e4d601b82b19a2fed9e612671a6296d51RUN
    ln -s /opt/tritonserver/backends/pytorch/* /usr/local/lib/
    12/31/2022 1:47 AM UTC
    sha256:d682c3f12ab242ae2e5adaa4b367988d93348ebdd00fbeb8e49f11fb545c1421COPY
    --chown=1000:1000dir:78caef98c724c087bc49df720cca19af83458677e6e9b4ba3be341187d03511b in /usr/local/lib/python3.8/dist-packages/torch
    12/31/2022 1:47 AM UTC
    sha256:8758332a2f2e0eaee15165a685c38f9c41936a50e4f1e1b1546c36ae54d9a90fCOPY
    --chown=1000:1000dir:3b74f563a5b3bcb3371739465e8068e0ea2866d7cf613d60c13c3352d49fe32c in /usr/local/lib/python3.8/dist-packages/numpy
    12/31/2022 1:47 AM UTC
    sha256:d3d31f076adabe64d1f1c81e48dc84315bb6ac657e47a94f7a4f08b39b289709COPY
    --chown=1000:1000dir:321b4265471393fd4d1240e6e3ba5037b21ab0161389bc133a9b4d3a7491a7fc in /usr/local/lib/python3.8/dist-packages/numba
    12/31/2022 1:47 AM UTC
    sha256:aed1c7de1b9cb6e7dcbc366476b884abd77d15675561befdc9320fd2af2f1c79COPY
    --chown=1000:1000dir:6e6aff9279812ba5ea0ecd4129555ab104f2d0c5cd4692269a2d63d14120a88a in backends/pytorch
    12/31/2022 1:46 AM UTC
    sha256:6f21c5dc422fbcd319c749d8531cf9c35fe95415bb412bd21baa4f45fffd8c19RUN
    pip install --no-cache-dir --no-deps torch torchmetrics &&
      rm -rf /usr/local/lib/python3.8/dist-packages/torch &&
      rm -rf /usr/local/lib/python3.8/dist-packages/caffe2
    12/31/2022 1:46 AM UTC
    sha256:530b34a45ee7f9e95c275dd0742b6800fc78f17d32682aa47e964b280e94aa10RUN
    apt update -y --fix-missing &&
      apt install -y --no-install-recommends libatlas-base-dev libmkl-dev &&
      apt autoremove -y &&
      apt clean &&
      rm -rf /var/lib/apt/lists/*
    12/31/2022 1:46 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
    /opt/nvidia/nvidia_entrypoint.sh
    12/31/2022 12:57 AM UTC
    ...