add docker deployment

pull/312/head
daihui 2023-07-15 11:10:14 +08:00
parent 5486f4f170
commit 536971b487
2 changed files with 36 additions and 0 deletions

View File

@ -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]

16
docker/dockerfile Normal file
View File

@ -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