kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

update Dockerfile

Browse Source
This commit is contained in:
Qing 2022-11-01 09:09:43 +08:00
parent 7147ffac7f
commit db75357659
4 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
README.md
View File

@ -197,7 +197,7 @@ The cache directories for different models correspond as follows:
docker run -p 8080:8080 \ docker run -p 8080:8080 \
-v /path/to/torch_cache:/root/.cache/torch \ -v /path/to/torch_cache:/root/.cache/torch \
-v /path/to/huggingface_cache:/root/.cache/huggingface \ -v /path/to/huggingface_cache:/root/.cache/huggingface \
--rm lama-cleaner:cpu-0.21.0 \ --rm lama-cleaner:cpu-0.24.4 \
lama-cleaner --device=cpu --port=8080 --host=0.0.0.0 lama-cleaner --device=cpu --port=8080 --host=0.0.0.0
``` ```
@ -211,7 +211,7 @@ lama-cleaner --device=cpu --port=8080 --host=0.0.0.0
docker run --gpus all -p 8080:8080 \ docker run --gpus all -p 8080:8080 \
-v /path/to/torch_cache:/root/.cache/torch \ -v /path/to/torch_cache:/root/.cache/torch \
-v /path/to/huggingface_cache:/root/.cache/huggingface \ -v /path/to/huggingface_cache:/root/.cache/huggingface \
--rm lama-cleaner:cpu-0.21.0 \ --rm lama-cleaner:gpu-0.24.4 \
lama-cleaner --device=cuda --port=8080 --host=0.0.0.0 lama-cleaner --device=cuda --port=8080 --host=0.0.0.0
``` ```
@ -222,11 +222,11 @@ Then open [http://localhost:8080](http://localhost:8080)
cpu only cpu only
``` ```
docker build -f ./docker/CPUDockerfile -t lamacleaner . docker build -f --build-arg version=0.x.0 ./docker/CPUDockerfile -t lamacleaner .
``` ```
gpu & cpu gpu & cpu
``` ```
docker build -f ./docker/GPUDockerfile -t lamacleaner . docker build -f --build-arg version=0.x.0 ./docker/GPUDockerfile -t lamacleaner .
``` ```

4
build_docker.sh
View File

@ -1,7 +1,7 @@
#!/usr/bin/env bash #!/usr/bin/env bash
set -e set -e
GIT_TAG=`git describe --tags --abbrev=0` GIT_TAG=$1
IMAGE_DESC="Image inpainting tool powered by SOTA AI Model" IMAGE_DESC="Image inpainting tool powered by SOTA AI Model"
GIT_REPO="https://github.com/Sanster/lama-cleaner" GIT_REPO="https://github.com/Sanster/lama-cleaner"
@ -14,6 +14,7 @@ docker buildx build \
--label org.opencontainers.image.url=$GIT_REPO \ --label org.opencontainers.image.url=$GIT_REPO \
--label org.opencontainers.image.source=$GIT_REPO \ --label org.opencontainers.image.source=$GIT_REPO \
--label org.opencontainers.image.version=$GIT_TAG \ --label org.opencontainers.image.version=$GIT_TAG \
--build-arg version=$GIT_TAG \
--tag lama-cleaner:cpu-$GIT_TAG . --tag lama-cleaner:cpu-$GIT_TAG .
@ -26,4 +27,5 @@ docker buildx build \
--label org.opencontainers.image.url=$GIT_REPO \ --label org.opencontainers.image.url=$GIT_REPO \
--label org.opencontainers.image.source=$GIT_REPO \ --label org.opencontainers.image.source=$GIT_REPO \
--label org.opencontainers.image.version=$GIT_TAG \ --label org.opencontainers.image.version=$GIT_TAG \
--build-arg version=$GIT_TAG \
--tag lama-cleaner:gpu-$GIT_TAG . --tag lama-cleaner:gpu-$GIT_TAG .

5
docker/CPUDockerfile
View File

@ -7,7 +7,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
RUN pip install --upgrade pip && \ RUN pip install --upgrade pip && \
pip install torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cpu pip install torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cpu
RUN pip install lama-cleaner
ARG version
RUN pip install lama-cleaner==$version
EXPOSE 8080 EXPOSE 8080

7
docker/GPUDockerfile
View File

@ -6,8 +6,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
curl python3-pip curl python3-pip
RUN pip3 install --upgrade pip && \ RUN pip3 install --upgrade pip && \
pip3 install torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cu116 pip3 install torch==1.12.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
RUN pip3 install lama-cleaner
ARG version
RUN pip3 install lama-cleaner==$version
EXPOSE 8080 EXPOSE 8080