Как интегрировать Torch в образ докера, сохраняя при этом разумный размер изображения? ⇐ Python

[Anonymous]

Итак, у меня есть веб-приложение Flask, которое будет отображать некоторые модели глубокого обучения.
Я создал образ, и все работает нормально.
проблема в том, что размер этого образа составляет 5,58 ГБ! что немного смешно.
У меня есть несколько моделей глубокого обучения, которые копируются во время сборки. Я думал, что они могут быть виновниками, но их совокупный размер не превышает 300 МБ, поэтому это определенно не то.
проверив историю и размер каждого слоя, я обнаружил следующее:

• RUN /bin/sh -c pip install -r requirements.txt занимает 771 МБ.
• RUN /bin/sh -c pip install torch==1.10.2 занимает 2,8 ГБ!
• RUN /bin/sh -c apt-get install ffmpeg libsm6 libxext6 занимает 400 МБ.

так как мне подключить эти библиотеки, сохранив при этом разумный размер изображения? можно ли иметь изображения такого размера при развертывании моделей мл в Python?
ниже находится корневой каталог:


Dockerfile:
FROM python:3.7.13
WORKDIR /app
COPY ["rdm.pt", "autosort_model.pt", "rotated_model.pt", "yolov5x6.pt", "/app/"]
RUN pip install torch==1.10.2
COPY requirements.txt /app/requirements.txt
RUN pip install -r requirements.txt
RUN apt-get update
RUN apt-get install ffmpeg libsm6 libxext6 -y
COPY . /app
CMD python ./app.py

.dockerignore:
Dockerfile
README.md
__pycache__


Подробнее здесь: https://stackoverflow.com/questions/732 ... reasonable