如何配置Stable Diffusion服务器 - AI技术

配置 Stable Diffusion 服务器需要多个步骤，包括环境准备、模型下载、依赖安装和服务启动。以下是详细的配置指南，涵盖从零开始的完整流程：

一、环境准备

1. 硬件要求

GPU：推荐 NVIDIA 显卡（显存 ≥ 8GB，如 RTX 3060/3080/4090）。
最低显存：4GB（生成 512x512 图像可能需用 --medvram 或 --lowvram 参数）。
内存：≥ 16GB RAM。
存储：至少 50GB 可用空间（模型文件较大，如 v1.5 约 4GB，v2.1 约 6GB）。
CPU：多核 CPU（如 Intel i7/AMD Ryzen 7）。

2. 软件要求

操作系统：Windows 10/11、Linux（Ubuntu 20.04+）或 macOS（需适配）。
Python：3.10.x（推荐版本）。
CUDA：与 GPU 驱动兼容的版本（如 CUDA 11.8）。
Git：用于克隆代码仓库。

二、安装依赖

1. Windows 环境

安装 Python

下载 Python 3.10.x 并安装，勾选 Add Python to PATH。

安装 Git

下载 Git 并安装。

安装 NVIDIA 驱动和 CUDA

更新 NVIDIA 驱动至最新版：驱动下载。
安装 CUDA Toolkit 11.8（需与 PyTorch 兼容）。

2. Linux 环境（Ubuntu）

# 更新系统
sudo apt update && sudo apt upgrade -y
# 安装 Python 3.10
sudo apt install python3.10 python3.10-venv python3-pip -y
# 安装 Git
sudo apt install git -y
# 安装 NVIDIA 驱动（如果未安装）
sudo ubuntu-drivers autoinstall
sudo reboot
# 安装 CUDA（示例为 CUDA 11.8）
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda-repo-ubuntu2204-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2204-11-8-local_11.8.0-520.61.05-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2204-11-8-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt update
sudo apt install cuda -y

三、安装 Stable Diffusion WebUI

推荐使用 AUTOMATIC1111 的 WebUI（最流行的分支）：

# 克隆仓库
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui

1. 安装依赖

Windows

# 创建虚拟环境（可选但推荐）
python -m venv venv
venv\Scripts\activate
# 安装依赖
pip install -r requirements.txt

Linux/macOS

# 创建虚拟环境
python3 -m venv venv
source venv/bin/activate
# 安装依赖
pip install -r requirements.txt

四、下载模型

1. 获取模型文件

推荐模型：
Stable Diffusion v1.5：下载地址
Stable Diffusion v2.1：下载地址
其他模型：从 Hugging Face 或 Civitai 下载。

2. 放置模型文件

将下载的 .ckpt 或 .safetensors 文件放入：

stable-diffusion-webui/models/Stable-diffusion/

五、启动服务器

1. 运行 WebUI

Windows

# 进入目录
cd stable-diffusion-webui
# 启动（自动下载依赖）
webui-user.bat

Linux/macOS

# 进入目录
cd stable-diffusion-webui
# 启动（添加参数以优化性能）
./webui.sh --medvram --xformers

2. 常用启动参数

--medvram：中等显存优化（适合 6-8GB 显存）。
--lowvram：低显存优化（适合 4GB 显存）。
--xformers：启用 xFormers 加速（需显卡支持）。
--listen：允许局域网访问（需配置防火墙）。
--port 7860：指定端口（默认 7860）。

六、访问 WebUI

启动后，终端会显示类似以下信息：

Running on local URL:  http://127.0.0.1:7860

浏览器打开 http://127.0.0.1:7860 即可使用。

七、进阶配置

1. 远程访问

局域网：启动时添加 --listen，然后在其他设备访问 http://服务器IP:7860。
公网：使用反向代理（如 Nginx）或内网穿透工具（如 frp、ngrok）。

2. 多用户管理

使用 --gradio-auth 用户名:密码 设置访问权限：

./webui.sh --gradio-auth user1:pass123

3. 安装扩展

在 WebUI 的 Extensions 标签页中搜索并安装插件（如 ControlNet、Lora 训练）。

八、常见问题

1. 显存不足

使用 --medvram 或 --lowvram 参数。
降低生成分辨率（如 512x512）。

2. 依赖安装失败

更换 pip 镜像源：

pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

3. CUDA 错误

确保 CUDA 版本与 PyTorch 兼容，可重新安装 PyTorch：

pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118

完成以上步骤后，您已成功配置 Stable Diffusion 服务器！如需进一步优化或定制功能，可参考 AUTOMATIC1111 文档。