update Dockerfile

2022-11-01 09:09:43 +08:00
parent 7147ffac7f
commit db75357659
4 changed files with 18 additions and 10 deletions
--- a/docker/GPUDockerfile
+++ b/docker/GPUDockerfile
@@ -6,9 +6,12 @@ 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 --extra-index-url https://download.pytorch.org/whl/cu116
-RUN pip3 install lama-cleaner
+    pip3 install torch==1.12.1+cu116 --extra-index-url https://download.pytorch.org/whl/cu116
+    
+ARG version 
+
+RUN pip3 install lama-cleaner==$version

 EXPOSE 8080

-CMD ["bash"]
+CMD ["bash"]