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:61121e79a9b6ef75626df7d24f3cc4d574e6aaca90575339d301238da3659d8aRUN
    pip install matplotlib
    11/23/2022 5:12 PM UTC
    sha256:abf8119c28dec9a3d6bcc059ccde261b2d4c3e98b8b088d99585352868356b8dRUN
    pip install --no-deps fastai fastcore fastprogress fastdownload
    11/23/2022 5:11 PM UTC
    sha256:1dc67026a8496713481bdd8854ff12a190e6f6e5a883164f4cd36cc50d9fcac2COPY
    --chown=1000:1000dir:6f07987f8cb6c3fd3725beed0ea1a5861d4101e15c8f20c12e58d706191e86d8 in /usr/local/lib/python3.8/dist-packages/torch
    11/23/2022 5:11 PM UTC
    sha256:cf78854826afe64c9d8f3e769e1c39bb9fe4474af751fa5a74b4db6a48fdc9bdCOPY
    --chown=1000:1000dir:76386bde9f857a3d3b759e6831ae01ff0f76142feea94e09ab3e483616190c0a in /usr/local/lib/python3.8/dist-packages/numpy
    11/23/2022 5:11 PM UTC
    sha256:ebb1c35a0935e8c3d8a568fca8e7a6f4699337b4cf55182a69061971534e5657COPY
    --chown=1000:1000dir:226bfeb0fa603039a20a2b22903990a7a8daf610d08ef18f88505dce3e0d9903 in /usr/local/lib/python3.8/dist-packages/numba
    11/23/2022 5:11 PM UTC
    sha256:ed414a2f15e019a52ffd66f875009bb83466859d18a81cb8cabf191ae6610fc7COPY
    --chown=1000:1000dir:03e8d035af9fb9a9f9ea3cf98748c267ccca444c9d7bf3886ed3aca92dccbdb1 in backends/pytorch
    11/23/2022 5:11 PM UTC
    sha256:ff44c6972795241e8d0b0a99b80f62fd1e2b00aeeaf38c6b9bd89b0d97eff04bRUN
    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
    11/23/2022 5:11 PM UTC
    sha256:204802de7be369c4d78531760a53ede3bdf34a722250b31e8980f56d64c5feacRUN
    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/*
    11/23/2022 5:10 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
    /opt/nvidia/nvidia_entrypoint.sh
    11/23/2022 3:49 PM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
    /bin/bash
    11/23/2022 3:49 PM UTC
    ...