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

update dockerfile

Browse Source
This commit is contained in:
Qing 2023-01-20 09:30:14 +08:00
parent 00def97e9f
commit b35cffadbd
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
build_docker.sh
View File

@ -15,7 +15,7 @@ docker buildx build \
--label org.opencontainers.image.source=$GIT_REPO \
--label org.opencontainers.image.version=$GIT_TAG \
--build-arg version=$GIT_TAG \
--tag lama-cleaner:cpu-$GIT_TAG .
--tag cwq1913/lama-cleaner:cpu-$GIT_TAG .
echo "Building NVIDIA GPU docker image..."
@ -28,4 +28,4 @@ docker buildx build \
--label org.opencontainers.image.source=$GIT_REPO \
--label org.opencontainers.image.version=$GIT_TAG \
--build-arg version=$GIT_TAG \
--tag lama-cleaner:gpu-$GIT_TAG .
--tag cwq1913/lama-cleaner:gpu-$GIT_TAG .

2
docker/CPUDockerfile
View File

@ -6,7 +6,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
curl
RUN pip install --upgrade pip && \
pip install torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cpu
pip install torch==1.13.1 --extra-index-url https://download.pytorch.org/whl/cpu
ARG version

6
docker/GPUDockerfile
View File

@ -1,4 +1,4 @@
FROM nvidia/cuda:11.6.1-runtime-ubuntu20.04
FROM nvidia/cuda:11.7.1-runtime-ubuntu20.04
RUN apt-get update && apt-get install -y --no-install-recommends \
software-properties-common \
@ -6,8 +6,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
curl python3-pip
RUN pip3 install --upgrade pip && \
pip3 install torch==1.12.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
pip3 install torch==1.13.1 xformers==0.0.16rc425
ARG version
RUN pip3 install lama-cleaner==$version