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:fc6401c3b41ee563408f07b32d461232528a753d82797181c210f69d28fa0cc7RUN
    pip install matplotlib
    10/27/2022 5:36 AM UTC
    sha256:577ab49bd156326cee73e59156078b97c4b9f28fc25ec649e2cbac69ca22327eRUN
    pip install --no-deps fastai fastcore fastprogress fastdownload
    10/27/2022 5:36 AM UTC
    sha256:559e0653bf885565c5f8f6d17d45e47e2ef03721ce0143d0db6dff2d4649bd81COPY
    --chown=1000:1000dir:6f07987f8cb6c3fd3725beed0ea1a5861d4101e15c8f20c12e58d706191e86d8 in /usr/local/lib/python3.8/dist-packages/torch
    10/27/2022 5:36 AM UTC
    sha256:39c58d36373eda1374d261fff89ca3b649862bc3fa9c1dc11751c13bf0d75c90COPY
    --chown=1000:1000dir:76386bde9f857a3d3b759e6831ae01ff0f76142feea94e09ab3e483616190c0a in /usr/local/lib/python3.8/dist-packages/numpy
    10/27/2022 5:36 AM UTC
    sha256:9becb12fa2cf2e5379bf293bbb464178811a55a40b9790df908db2f4feaff864COPY
    --chown=1000:1000dir:226bfeb0fa603039a20a2b22903990a7a8daf610d08ef18f88505dce3e0d9903 in /usr/local/lib/python3.8/dist-packages/numba
    10/27/2022 5:36 AM UTC
    sha256:8e94983068e2cd9230fcb07857573f0c2e5322550dfee3f4652c282b6287b61dCOPY
    --chown=1000:1000dir:03e8d035af9fb9a9f9ea3cf98748c267ccca444c9d7bf3886ed3aca92dccbdb1 in backends/pytorch
    10/27/2022 5:36 AM UTC
    sha256:b657a9049e1d309c24c6f7ef0763e16623ecfec351ffdfb20602a40961168e05RUN
    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
    10/27/2022 5:35 AM UTC
    sha256:a2145b56bfb67eaede286983fb4fd5753577307d1f19a18a6b446b7dd59eefd7RUN
    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/*
    10/27/2022 5:35 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4ENTRYPOINT
    /opt/nvidia/nvidia_entrypoint.sh
    10/26/2022 6:42 AM UTC
    sha256:a3ed95caeb02ffe68cdd9fd84406680ae93d633cb16422d00e8a7c22955b46d4CMD
    /bin/bash
    10/26/2022 6:42 AM UTC
    ...