增加docker部署配置

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_demo.py"
+ deploy:
+ resources:
+ reservations:
+ devices:
+ - driver: nvidia
+ count: 1
+ capabilities: [gpu]

docker/dockerfile

@@ -0,0 +1,18 @@
+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 .requirements .
+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 --no-cache-dir
+ADD . .
+# preinstall model
+RUN python -c "from transformers import AutoTokenizer, AutoModel; AutoModel.from_pretrained('THUDM/chatglm2-6b', trust_remote_code=True)"