第六十九篇-NVIDIA V100-32G+Stable-Diffusion-WebUI

环境

复制代码

系统：CentOS-7
CPU : E5-2680V4 14核28线程
内存：DDR4 2133 32G * 2
显卡：Tesla V100-32G【PG503】 (水冷)
驱动: 535
CUDA: 12.2

创建目录

复制代码

mkdir -p /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/data
mkdir -p /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/models/Stable-diffusion
mkdir -p /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/models/VAE
mkdir -p /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/models/Lora
mkdir -p /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/models/ControlNet
mkdir -p /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/models/ESRGAN
mkdir -p /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/outputs
mkdir -p /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/config
mkdir -p /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/extensions

cd /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/
chown 10000:$UID -R models outputs
chmod 775 -R models outputs

基于自己目录调整

下载镜像

复制代码

docker pull docker.1ms.run/siutin/stable-diffusion-webui-docker:latest-cuda-12.2.2
基于自己cuda版本选择

创建运行容器

复制代码

docker run -d --name sd-webui-2 \
  --gpus all \
  --network host \
  -v /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/data:/stable-diffusion-webui/data \
  -v /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/models:/app/stable-diffusion-webui/models \
  -v /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/outputs:/app/stable-diffusion-webui/outputs \
  -v /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/config:/app/stable-diffusion-webui/config \
  -v /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/extensions:/app/stable-diffusion-webui/extensions \
  -e CLI_ARGS="--xformers --enable-insecure-extension-access --listen --port 7860" \
  docker.1ms.run/siutin/stable-diffusion-webui-docker:latest-cuda-12.2.2 \
  bash webui.sh --share
	
	7860可以调整

进入容器

复制代码

docker exec -it sd-webui bash

增加export到 webui-user.sh

复制代码

sed -i '1i\export COMMANDLINE_ARGS="--listen --port 7860"\nexport HF_ENDPOINT=https://hf-mirror.com' webui-user.sh

重启容器

复制代码

docker restart sd-webui

下载模型

java 复制代码

https://hf-mirror.com/stable-diffusion-v1-5/stable-diffusion-v1-5/blob/main/v1-5-pruned-emaonly.ckpt
	放入宿主机 /opt/ai-runner/sd-webui-docker/stable-diffusion-webui/models/Stable-diffusion

访问

http://宿主机IP:7860