update dockerfile

This commit is contained in:
Qing 2023-04-16 21:07:20 +08:00
parent c8f0359c8c
commit 05b3514f1c

View File

@ -1,16 +1,18 @@
FROM python:3.7.4-slim-buster FROM python:3.10.11-slim-buster
RUN apt-get update && apt-get install -y --no-install-recommends \ RUN apt-get update && apt-get install -y --no-install-recommends \
software-properties-common \ software-properties-common \
libsm6 libxext6 ffmpeg libfontconfig1 libxrender1 libgl1-mesa-glx \ libsm6 libxext6 ffmpeg libfontconfig1 libxrender1 libgl1-mesa-glx \
curl curl gcc build-essential
RUN pip install --upgrade pip && \ RUN pip install --upgrade pip && \
pip install torch==1.13.1 --extra-index-url https://download.pytorch.org/whl/cpu pip install torch==1.13.1 torchvision==0.14.1 --extra-index-url https://download.pytorch.org/whl/cpu
ARG version ARG version
RUN pip install lama-cleaner==$version RUN pip install lama-cleaner==$version
RUN lama-cleaner --install-plugins-package
ENV LD_PRELOAD=/usr/local/lib/python3.10/site-packages/skimage/_shared/../../scikit_image.libs/libgomp-d22c30c5.so.1.0.0
EXPOSE 8080 EXPOSE 8080