终极指南：如何构建最小化的Stable Diffusion WebUI Docker镜像

终极指南：如何构建最小化的Stable Diffusion WebUI Docker镜像

【免费下载链接】stable-diffusion-webui-dockerEasy Docker setup for Stable Diffusion with user-friendly UI项目地址: https://gitcode.com/gh_mirrors/st/stable-diffusion-webui-docker

Stable Diffusion WebUI Docker是一个简单易用的Docker配置项目，帮助用户快速搭建带有友好界面的Stable Diffusion环境。本指南将详细介绍如何构建最小化的Stable Diffusion WebUI Docker镜像，让你的AI绘图体验更加高效流畅。

为什么选择Docker构建Stable Diffusion WebUI？

使用Docker构建Stable Diffusion WebUI具有诸多优势：

• 🚀环境一致性：确保在不同机器上运行时拥有相同的环境配置
• 🛡️隔离性：避免与系统其他软件产生冲突
• 📦便携性：轻松在不同设备间迁移
• 🔧简化部署：一键启动，无需复杂配置

准备工作

在开始构建之前，请确保你的系统已经安装了Docker和Docker Compose。项目的完整代码可以通过以下命令获取：

git clone https://gitcode.com/gh_mirrors/st/stable-diffusion-webui-docker

理解项目结构

项目的核心结构如下：

• docker-compose.yml：定义服务组合和配置
• services/：包含不同WebUI的Docker配置
  • AUTOMATIC1111/：AUTOMATIC1111版本WebUI的Docker配置
  • comfy/：ComfyUI的Docker配置
  • download/：模型下载服务配置

构建最小化镜像的关键步骤

1. 优化基础镜像选择

在services/AUTOMATIC1111/Dockerfile中，项目使用了PyTorch官方镜像作为基础：

FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime

这个基础镜像已经包含了运行Stable Diffusion所需的大部分依赖，但我们可以通过以下方式进一步优化：

• 考虑使用更小的基础镜像，如Alpine版本（如果兼容性允许）
• 只安装必要的系统依赖，避免臃肿

2. 精简构建步骤

Dockerfile中的构建步骤可以通过以下方式优化：

• 合并RUN命令，减少镜像层数
• 使用--mount=type=cache缓存依赖安装，如项目中已使用的：

RUN --mount=type=cache,target=/root/.cache/pip \ pip install pyngrok xformers==0.0.26.post1 \ git+https://github.com/TencentARC/GFPGAN.git@8d2447a2d918f8eba5a4a01463fd48e45126a379 \ git+https://github.com/openai/CLIP.git@d50d76daa670286dd6cacf3bcd80b5e4823fc8e1 \ git+https://github.com/mlfoundations/open_clip.git@v2.20.0

3. 优化模型下载

项目提供了专门的下载服务，通过services/download/download.sh脚本下载所需模型：

aria2c -x 10 --disable-ipv6 --input-file /docker/links.txt --dir /data/models --continue

为了最小化镜像大小，模型文件应存储在外部卷中，而不是包含在镜像内。在docker-compose.yml中，通过卷挂载实现这一点：

volumes: - &v1 ./data:/data - &v2 ./output:/output

4. 配置运行参数

在docker-compose.yml中，可以通过环境变量CLI_ARGS配置WebUI的运行参数，选择适合你硬件的配置：

environment: - CLI_ARGS=--allow-code --medvram --xformers --enable-insecure-extension-access --api

对于低配置系统，可以使用--medvram（中等显存模式）或--lowvram（低显存模式）来减少内存占用。

构建和运行最小化镜像

完成上述优化后，使用以下命令构建并启动服务：

# 下载所需模型 docker-compose --profile download up # 构建并启动AUTOMATIC1111 WebUI docker-compose --profile auto up --build

验证最小化镜像

构建完成后，可以通过以下命令检查镜像大小：

docker images | grep sd-auto

你会发现优化后的镜像体积显著减小，同时保持了完整的功能。

总结

通过本文介绍的方法，你可以构建一个功能完备且体积最小的Stable Diffusion WebUI Docker镜像。这不仅节省了存储空间，还提高了部署和运行效率。无论你是AI绘图爱好者还是专业开发者，这个优化后的Docker配置都能为你提供流畅的使用体验。

如果你想尝试其他WebUI版本，可以使用类似的方法优化services/comfy/Dockerfile，并通过以下命令启动：

docker-compose --profile comfy up --build

享受你的AI绘图之旅吧！

【免费下载链接】stable-diffusion-webui-dockerEasy Docker setup for Stable Diffusion with user-friendly UI项目地址: https://gitcode.com/gh_mirrors/st/stable-diffusion-webui-docker