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:79e1c2016ee8bc240ab79851ccc17ee610a95227e0fa3262d3e96bc21ddc0bf7RUN
    pip install matplotlib
    07/27/2022 5:36 AM UTC
    sha256:92a05b8d0085847b5bd0c51bff768e177d2924708e26d0227f05a1d93d8899d5RUN
    pip install --no-deps fastai fastcore fastprogress fastdownload
    07/27/2022 5:36 AM UTC
    sha256:b2d63f7e75d03a620f77659bb74aad1f0d79750f3dbe8cca50bf2400e0c037b1COPY
    --chown=1000:1000dir:71f0a7b2880dd910f2e1dfa60e2e44e8ddaf033115d8dba19cdb628052afcf84 in /usr/local/lib/python3.8/dist-packages/torch
    07/27/2022 5:36 AM UTC
    sha256:c1ccda5fa848afe10f4a12061d1540d244baf40c66cca9013da458162306fd88COPY
    --chown=1000:1000dir:ab5ce33d27a23562798b929a6711c70f6e22f419cde4c5f0da6814042596bd21 in /usr/local/lib/python3.8/dist-packages/numpy
    07/27/2022 5:36 AM UTC
    sha256:a743065111414ef4830bd83623f90a33a6634082cd40582156db940bbffe40ebCOPY
    --chown=1000:1000dir:254113775845405af1730557c3e681d3daa187225cd890a8567dd4341fb6b63d in /usr/local/lib/python3.8/dist-packages/numba
    07/27/2022 5:36 AM UTC
    sha256:4ecf6563a01df555fc39669fc003e4d109d85b00a82a583b9e9e80149e660df0COPY
    --chown=1000:1000dir:b3170a86e45dc73f932bb1984479f1a02c341bf2f712e83957c90e9af17be4cc in backends/pytorch
    07/27/2022 5:36 AM UTC
    sha256:595bef43756edd08c51f7431ec7a50dab8551ae0e429476b663a7680202735c3RUN
    pip install --no-cache-dir --no-deps torch torchmetrics &&
      rm -r /usr/local/lib/python3.8/dist-packages/torch &&
      rm -r /usr/local/lib/python3.8/dist-packages/caffe2
    07/27/2022 5:36 AM UTC
    sha256:b30b77dcb43be072420fed820d3aba7882f3029723ae998e19ff83802e6a263cRUN
    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/*
    07/27/2022 5:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
    /opt/nvidia/nvidia_entrypoint.sh
    07/26/2022 8:46 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
    /bin/bash
    07/26/2022 8:46 AM UTC
    ...