mirror of
https://github.com/immich-app/immich.git
synced 2024-12-26 10:50:29 +02:00
ab90b01122
* 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
20 lines
493 B
Docker
20 lines
493 B
Docker
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"]
|