From 536971b4873731c65469cc318bbfb532410d4f66 Mon Sep 17 00:00:00 2001
From: daihui <daihu@qq.com>
Date: Sat, 15 Jul 2023 11:10:14 +0800
Subject: [PATCH] add docker deployment

---
 docker/docker--compose.yml | 20 ++++++++++++++++++++
 docker/dockerfile          | 16 ++++++++++++++++
 2 files changed, 36 insertions(+)
 create mode 100644 docker/docker--compose.yml
 create mode 100644 docker/dockerfile

diff --git a/docker/docker--compose.yml b/docker/docker--compose.yml
new file mode 100644
index 0000000..720af4f
--- /dev/null
+++ b/docker/docker--compose.yml
@@ -0,0 +1,20 @@
+version: "3"
+services:
+  stable-diffusion:
+    build:
+      dockerfile: ./Dockerfile
+      context: .
+    runtime: nvidia
+    environment:
+      - NVIDIA_VISIBLE_DEVICES=all
+    working_dir: /app
+    ports:
+      - "9000:8000"
+    command: bash -c "python web_demo2.py"
+    deploy:
+      resources:
+        reservations:
+          devices:
+          - driver: nvidia
+            count: 1
+            capabilities: [gpu]
diff --git a/docker/dockerfile b/docker/dockerfile
new file mode 100644
index 0000000..15ae7f7
--- /dev/null
+++ b/docker/dockerfile
@@ -0,0 +1,16 @@
+FROM nvidia/cuda:11.7.1-runtime-ubuntu22.04
+
+RUN sed -i s/archive.ubuntu.com/mirrors.aliyun.com/g /etc/apt/sources.list
+RUN sed -i s/security.ubuntu.com/mirrors.aliyun.com/g /etc/apt/sources.list
+RUN sed -i s/ports.ubuntu.com/mirrors.aliyun.com/g /etc/apt/sources.list
+RUN apt update && apt-get -y install git wget \
+    python3.10 python3.10-venv python3-pip \
+    build-essential libgl-dev libglib2.0-0 wget
+RUN ln -s /usr/bin/python3.10 /usr/bin/python
+RUN useradd -ms /bin/bash usera
+WORKDIR /app
+ADD . .
+RUN pip config set global.index-url https://mirrors.cloud.tencent.com/pypi/simple
+RUN pip config set global.trusted-host mirrors.cloud.tencent.com
+RUN pip install -r requirements.txt
+