immich/machine-learning/Dockerfile

FROM python:3.10 as builder

ENV PYTHONDONTWRITEBYTECODE=1 \
  PYTHONUNBUFFERED=1 \
  PIP_NO_CACHE_DIR=true

RUN python -m venv /opt/venv
RUN /opt/venv/bin/pip install torch --index-url https://download.pytorch.org/whl/cpu
RUN /opt/venv/bin/pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece fastapi Pillow uvicorn[standard]
RUN /opt/venv/bin/pip install --no-deps sentence-transformers
# Facial Recognition Stuff
RUN /opt/venv/bin/pip install insightface onnxruntime

FROM python:3.10-slim

ENV NODE_ENV=production

COPY --from=builder /opt/venv /opt/venv

ENV TRANSFORMERS_CACHE=/cache \
  PYTHONDONTWRITEBYTECODE=1 \
  PYTHONUNBUFFERED=1 \
  PATH="/opt/venv/bin:$PATH"

WORKDIR /usr/src/app

COPY . .
ENV PYTHONPATH=`pwd`
CMD ["python", "src/main.py"]
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-03 22:45:20 +00:00			`FROM python:3.10 as builder`
fix(server): Flask not found (#1830) * fix(server): Flask not found * trial 1 2023-02-22 16:53:59 +00:00
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-03 22:45:20 +00:00			`ENV PYTHONDONTWRITEBYTECODE=1 \`
chore(ml): move to fastAPI (#2336) 2023-04-26 10:39:24 +00:00			`PYTHONUNBUFFERED=1 \`
			`PIP_NO_CACHE_DIR=true`
feat(server): Machine learning's image optimisations (#1908) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable 2023-03-01 15:37:12 +00:00
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 17:48:35 +00:00			`RUN python -m venv /opt/venv`
chore(ml): use official pytorch channel (#2416) 2023-05-10 12:09:33 +00:00			`RUN /opt/venv/bin/pip install torch --index-url https://download.pytorch.org/whl/cpu`
chore(ml): move to fastAPI (#2336) 2023-04-26 10:39:24 +00:00			`RUN /opt/venv/bin/pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece fastapi Pillow uvicorn[standard]`
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-03 22:45:20 +00:00			`RUN /opt/venv/bin/pip install --no-deps sentence-transformers`
feat: facial recognition (#2180) 2023-05-17 17:07:17 +00:00			`# Facial Recognition Stuff`
			`RUN /opt/venv/bin/pip install insightface onnxruntime`
chore(ml): move to fastAPI (#2336) 2023-04-26 10:39:24 +00:00
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-03 22:45:20 +00:00			`FROM python:3.10-slim`

chore(docker): Default NODE_ENV to production for server image and update alpine version (#2157) * default NODE_ENV to production for server image * update node image to use 3.17 alpine in server * update web docker image to use alpine 3.17 * remove NODE_ENV from production docker-compose * NODE_ENV is also needed default in machine-learning 2023-04-03 20:05:29 +00:00			`ENV NODE_ENV=production`

Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-03 22:45:20 +00:00			`COPY --from=builder /opt/venv /opt/venv`
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 17:48:35 +00:00
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-03 22:45:20 +00:00			`ENV TRANSFORMERS_CACHE=/cache \`
chore(ml): move to fastAPI (#2336) 2023-04-26 10:39:24 +00:00			`PYTHONDONTWRITEBYTECODE=1 \`
			`PYTHONUNBUFFERED=1 \`
			`PATH="/opt/venv/bin:$PATH"`
Feat/ml image optimisations (#1916) * Use multi stage build to slim down ML image size * Use gunicorn as WSGI server in ML image * Configure gunicorn server for ML use case * Use requirements.txt file to install python dependencies in ML image * Make ML listen IP configurable * Revert "Use requirements.txt file to install python dependencies in ML image" This reverts commit 32e706c7f3db1d0aa1a66b3733ab9ec9dfb86534. * Separate out pip installs in ML builder image 2023-03-03 22:45:20 +00:00
			`WORKDIR /usr/src/app`
Revert "feat(server): Machine learning's image optimisations (#1908)" (#1915) This reverts commit 977740045a6c1c29ea5edc4f22ad9377523c422d. 2023-03-01 17:48:35 +00:00
Added machine learning microservice and object detection (#76) 2022-03-27 19:58:54 +00:00			`COPY . .`
chore(ml): move to fastAPI (#2336) 2023-04-26 10:39:24 +00:00			ENV PYTHONPATH=`pwd`
chore(ml): Fix entrypoint path (#2373) 2023-05-03 19:27:35 +00:00			`CMD ["python", "src/main.py"]`