Files
geovisio-sgblur/Dockerfile
Roman Vanicek 2f054a19f8
All checks were successful
continuous-integration/drone/push Build is passing
Fix torchvision version
2024-03-25 23:14:14 +01:00

48 lines
1.2 KiB
Docker

FROM python:3.10-slim
WORKDIR /opt/blur
# Dependencies
RUN apt-get -qq update && DEBIAN_FRONTEND=noninteractive apt-get -y install \
# * Pillow
libffi-dev \
libfreetype6-dev \
libfribidi-dev \
libharfbuzz-dev \
libjpeg-turbo-progs \
libjpeg62-turbo-dev \
liblcms2-dev \
libopenjp2-7-dev \
libtiff5-dev \
libwebp-dev \
libssl-dev \
meson \
netpbm \
xvfb \
zlib1g-dev \
# * sgblur
libturbojpeg0-dev \
libjpeg-turbo-progs \
exiftran \
&& rm -rf /var/lib/apt/lists/*
COPY ./requirements.txt ./
RUN pip install -r ./requirements.txt && \
# Install torch and torchvision CPU only versions
pip install torch==2.2.1+cpu torchvision==0.17.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html && \
#Explicitly install ultralytics without optional dependencies (like CUDA). (see https://github.com/ultralytics/ultralytics/issues/749)
pip install ultralytics==8.1.19 --no-deps && \
pip cache purge
# Source files
COPY ./src ./src
COPY ./scripts ./scripts
COPY ./models ./models
COPY ./demo.html ./
COPY ./docker/docker-entrypoint.sh ./
RUN chmod +x ./docker-entrypoint.sh
# Expose service
EXPOSE 8001
ENTRYPOINT ["./docker-entrypoint.sh"]