kestr3l/stable-diffusion-webui Docker 镜像 - 轩辕镜像

Stable-Diffusion-Webui Docker 镜像文档

镜像概述与主要用途

本镜像基于 AUTOMATIC1111/stable-diffusion-webui 项目构建，提供 Docker 化部署的 Stable Diffusion Web UI 环境。镜像基于 nvidia/cuda 的 runtime-ubuntu22.04 变体，支持 GPU 加速，旨在简化 Stable Diffusion Web UI 的部署与管理。

• 源码仓库：kestr31/docker-stable-diffusion-webui
• 预构建镜像：kestr3l/stable-diffusion-webui

核心功能与特性

• GPU 加速支持：基于 nvidia/cuda 基础镜像，需配合 NVIDIA 容器工具包实现 GPU 资源调用。
• 便捷部署脚本：提供 build.sh 构建脚本与 sd-webui.sh 运行脚本，简化镜像构建与容器管理流程。
• 自定义配置：支持修改访问端口、添加 Web UI 命令行参数，满足个性化需求。
• 调试模式：支持以调试模式启动容器，便于问题排查与环境调试。
• 数据持久化：默认将工作区目录（含配置文件、模型数据等）存储在本地，确保数据不丢失。

使用场景与适用范围

适用环境

• Linux 操作系统或 WSL2 环境
• 配备 NVIDIA 显卡（需支持 CUDA）
• 已安装 Docker Engine、Docker Compose 及 NVIDIA Container Toolkit

适用用户

• 需要通过 Docker 快速部署 Stable Diffusion Web UI 的开发者
• 希望简化环境配置与依赖管理的 AI 绘画爱好者
• 需在隔离环境中运行 Stable Diffusion Web UI 的用户

前提条件

在使用本镜像前，需确保满足以下条件：

1. 操作系统：Linux 或 WSL2
2. 硬件：支持 CUDA 的 NVIDIA GPU
3. 软件依赖：
   • Docker Engine
   • Docker Compose（通常随 Docker Engine 一同安装）
   • NVIDIA Container Toolkit
4. 基础知识：熟悉 Linux 命令行操作

构建与运行

构建镜像

使用项目提供的 build.sh 脚本构建自定义镜像，脚本位于 scripts/ 目录下。

bash
./scripts/build.sh <镜像名称> <CUDA 版本>
# 示例：./scripts/build.sh kestr3l/stable-diffusion-webui 12.2.2
# 构建结果：生成镜像 "kestr3l/stable-diffusion-webui:1.3.0-12.2.2"

注意：脚本中 ${REPO_VERSION} 变量为硬编码，构建时会自动附加到镜像标签中。

运行容器

通过 sd-webui.sh 脚本部署容器，默认工作区目录为 ${HOME}/Documents/sd-webui（可自定义）。脚本会自动复制 compose.yml 与 run.env 至工作区，并克隆 stable-diffusion-webui 源码。

bash
./scripts/sd-webui.sh run <工作区目录（可选）>
# 示例：./scripts/sd-webui.sh run
# 效果：在 ${HOME}/Documents/sd-webui 创建工作区，部署并启动容器

• 访问 Web UI：容器启动后，通过 http://localhost:8000 访问（默认端口可修改）。
• 安全提示：默认启用 --listen 参数允许外部访问，建议立即配置防火墙或设置密码。
• 容器名称：默认容器名称为 sd-webui（在 run.env 中硬编码），CUDA 版本与 Web UI 版本也通过 run.env 硬编码指定。

停止容器

使用 sd-webui.sh 脚本的 stop 参数停止运行中的容器：

bash
./scripts/sd-webui.sh stop

调试容器

通过 debug 参数以调试模式启动容器，此时入口命令被覆盖为 sleep infinity，可进入容器内部排查问题：

bash
./scripts/sd-webui.sh debug
# 进入容器：docker exec -it sd-webui /bin/bash

自定义配置

修改访问端口

通过修改工作区目录下的 run.env 文件调整 Web UI 访问端口：

bash
# ${HOME}/Documents/sd-webui/run.env
...
WEBUI_PORT=7860  # 将端口修改为 7860
...

添加 Web UI 命令行参数

Web UI 的核心配置文件为 webui-user.sh，位于工作区的 stable-diffusion-webui 目录下（默认路径：${HOME}/Documents/sd-webui/stable-diffusion-webui）。通过修改该文件中的 COMMANDLINE_ARGS 变量添加自定义参数。

默认参数：--listen --enable-insecure-extension-access

示例：添加认证与加速参数

bash
# ${HOME}/Documents/sd-webui/stable-diffusion-webui/webui-user.sh
...
export COMMANDLINE_ARGS="--gradio-auth 用户名:密码 --xformers"  # 设置认证与 xFormers 加速
...

更多参数参考 AUTOMATIC1111 官方文档。

未来计划

• 实现 CI/CD 自动化镜像更新
• 提供训练环境专用镜像

参考资料

1. AUTOMATIC1111/stable-diffusion-webui（Web UI 核心项目）