Docker support for CPU and CUDA

s0md3v · Aug 14, 2023 · a83e408 · a83e408
1 parent fb9372c
commit a83e408
Show file tree

Hide file tree

Showing 4 changed files with 48 additions and 0 deletions.
diff --git a/Dockerfile.cpu b/Dockerfile.cpu
@@ -0,0 +1,11 @@
+FROM python:3.10
+
+ENV GRADIO_SERVER_NAME=0.0.0.0
+WORKDIR /roop
+
+RUN apt-get update
+RUN apt-get install -y ffmpeg
+
+RUN git clone https://github.com/s0md3v/roop.git .
+RUN git checkout next
+RUN pip install -r requirements-docker.cpu.txt
diff --git a/Dockerfile.cuda b/Dockerfile.cuda
@@ -0,0 +1,12 @@
+FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
+FROM python:3.10
+
+ENV GRADIO_SERVER_NAME=0.0.0.0
+WORKDIR /roop
+
+RUN apt-get update
+RUN apt-get install -y ffmpeg
+
+RUN git clone https://github.com/s0md3v/roop.git .
+RUN git checkout next
+RUN pip install -r requirements-docker.cuda.txt
diff --git a/docker-compose.cpu.yml b/docker-compose.cpu.yml
@@ -0,0 +1,9 @@
+services:
+ roop-cpu:
+ build:
+ dockerfile: Dockerfile.cpu
+ command: ['python', 'run.py']
+ volumes:
+ - ./models:/roop/models
+ ports:
+ - 8000:7860
diff --git a/docker-compose.cuda.yml b/docker-compose.cuda.yml
@@ -0,0 +1,16 @@
+services:
+ roop-cuda:
+ build:
+ dockerfile: Dockerfile.cuda
+ command: ['python', 'run.py --execution-provider cuda']
+ volumes:
+ - ./models:/roop/models
+ expose:
+ - 8000:7860
+ deploy:
+ resources:
+ reservations:
+ devices:
+ - driver: nvidia
+ count: 1
+ capabilities: [gpu, video]