immich/machine-learning/Dockerfile

FROM python:3.10

ENV TRANSFORMERS_CACHE=/cache \
    PYTHONDONTWRITEBYTECODE=1 \
    PYTHONUNBUFFERED=1 \
    PIP_NO_CACHE_DIR=true

WORKDIR /usr/src/app

RUN python -m venv /opt/venv
ENV PATH="/opt/venv/bin:$PATH"

RUN pip install --pre torch  -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
RUN pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece flask Pillow
RUN pip install --no-deps sentence-transformers

COPY . .

CMD ["python", "src/main.py"]
feat(machine-learning)!: move machine learning to Python based image (#1774) BREAKING CHANGES * Users have to update the docker-compose file, machine-learning portion. * Temporary dropping machine-learning support for Arm64 and Armv7 2023-02-18 15:13:37 +00:00			`FROM python:3.10`
Added machine learning microservice and object detection (#76) 2022-03-27 19:58:54 +00:00
feat(server) Enable ML support for ARM CPUs (#1880) * Install nightly release of pytorch to enable ML support for arm CPUs * Remove linux/arm/v7 from ML docker builds * Add --no-cache-dir to torch installation command in ML image build * Use PIP_NO_CACHE_DIR option in ML build to further decrease image size 2023-02-26 21:00:09 +00:00			`ENV TRANSFORMERS_CACHE=/cache \`
			`PYTHONDONTWRITEBYTECODE=1 \`
			`PYTHONUNBUFFERED=1 \`
			`PIP_NO_CACHE_DIR=true`
Added machine learning microservice and object detection (#76) 2022-03-27 19:58:54 +00:00
			`WORKDIR /usr/src/app`

fix(server): Flask not found (#1830) * fix(server): Flask not found * trial 1 2023-02-22 16:53:59 +00:00			`RUN python -m venv /opt/venv`
			`ENV PATH="/opt/venv/bin:$PATH"`

feat(server) Enable ML support for ARM CPUs (#1880) * Install nightly release of pytorch to enable ML support for arm CPUs * Remove linux/arm/v7 from ML docker builds * Add --no-cache-dir to torch installation command in ML image build * Use PIP_NO_CACHE_DIR option in ML build to further decrease image size 2023-02-26 21:00:09 +00:00			`RUN pip install --pre torch -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html`
fix(server): Flask not found (#1830) * fix(server): Flask not found * trial 1 2023-02-22 16:53:59 +00:00			`RUN pip install transformers tqdm numpy scikit-learn scipy nltk sentencepiece flask Pillow`
			`RUN pip install --no-deps sentence-transformers`
Added machine learning microservice and object detection (#76) 2022-03-27 19:58:54 +00:00
			`COPY . .`
fix(machine-learning): Add the command to execute at startup (#1843) * fix(machine-learning): Add the command to execute at startup Previously it wasn't set in the Docker container but it should be. * fix(docker): remove machine-learning command arg * fix(docker): machine-learning CMD argument 2023-02-23 15:54:04 +00:00
			`CMD ["python", "src/main.py"]`