From 05b3514f1ca4b43b28ac911fe956add2af40df31 Mon Sep 17 00:00:00 2001
From: Qing <cwq1913@gmail.com>
Date: Sun, 16 Apr 2023 21:07:20 +0800
Subject: [PATCH] update dockerfile

---
 docker/CPUDockerfile | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/docker/CPUDockerfile b/docker/CPUDockerfile
index 9304267..a314cd3 100644
--- a/docker/CPUDockerfile
+++ b/docker/CPUDockerfile
@@ -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 \
     software-properties-common \
     libsm6 libxext6 ffmpeg libfontconfig1 libxrender1 libgl1-mesa-glx \
-    curl
+    curl gcc build-essential
 
 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
 
 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