CentOS安装Stable Diffusion系统 - AI技术

CentOS安装Stable Diffusion系统指南

一、前置准备

1. 系统要求

操作系统：CentOS 7及以上（建议使用CentOS Stream或RHEL衍生版本以保证稳定性）；
硬件配置：至少8GB内存（推荐16GB+）、20GB以上可用存储（SSD优先）、支持CUDA的NVIDIA GPU（如GTX 1060及以上，需安装对应驱动）；
网络：稳定的互联网连接（用于下载模型及依赖）。

2. 依赖安装

更新系统并安装基础工具与库，确保后续步骤顺利进行：

sudo yum update -y
sudo yum groupinstall -y "Development Tools"  # 安装编译工具链
sudo yum install -y python3 python3-pip git cmake wget  # 核心依赖

二、可选但推荐的优化步骤

1. 创建Python虚拟环境

隔离项目环境，避免依赖冲突：

python3 -m venv stable-diffusion-env  # 创建虚拟环境
source stable-diffusion-env/bin/activate  # 激活环境
pip install --upgrade pip  # 升级pip至最新版本

2. 安装NVIDIA CUDA与cuDNN（GPU用户必做）

若使用NVIDIA GPU加速，需安装CUDA Toolkit（≥11.3）和cuDNN（≥8.2）：

CUDA安装：从NVIDIA官网下载对应版本的CUDA RPM包，通过rpm命令安装；
cuDNN安装：注册NVIDIA开发者账号，下载cuDNN包并解压至/usr/local/cuda目录；
环境变量配置：在~/.bashrc中添加：

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

执行source ~/.bashrc使配置生效。

三、部署Stable Diffusion

1. 安装PyTorch（GPU加速版）

根据CUDA版本选择对应的PyTorch安装命令（以CUDA 11.8为例）：

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

若使用CPU版本，可省略--index-url参数，但性能会显著下降。

2. 克隆Stable Diffusion仓库

选择官方或社区维护的仓库（如AUTOMATIC1111的WebUI）：

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui

3. 安装项目依赖

执行以下命令安装Python依赖库：

pip install -r requirements.txt

若遇到ImportError: cannot import name 'rank_zero_only'等报错，需手动修复：

pip install pytorch-lightning --upgrade  # 升级pytorch-lightning

4. 下载预训练模型

从Hugging Face模型库下载Stable Diffusion模型（如stable-diffusion-2-1）：

git lfs install  # 安装Git LFS（用于下载大文件）
git clone https://huggingface.co/stabilityai/stable-diffusion-2-1
cd stable-diffusion-2-1
git lfs pull  # 下载模型文件

模型文件较大（约5-10GB），需确保网络稳定。

四、运行Stable Diffusion

1. 启动WebUI

在项目根目录下执行以下命令启动Web服务：

python launch.py --listen  # 监听本地端口（默认7860）

若需指定GPU设备，可添加--gpu-index 0参数（多GPU用户需调整索引）。

2. 访问Web界面

本地访问：打开浏览器，输入http://localhost:7860；
远程访问：通过SSH反向隧道将本地端口映射至远程服务器（如ssh -L 7860:localhost:7860 user@server_ip），然后在浏览器输入http://localhost:7860。

五、常见问题解决

1. 内存不足

若系统内存不足，可创建交换空间（Swap）缓解：

sudo fallocate -l 10G /opt/swap  # 创建10GB交换文件
sudo chmod 600 /opt/swap
sudo mkswap /opt/swap
sudo swapon /opt/swap

验证交换空间是否启用：free -h。

2. 权限问题

若遇到权限拒绝错误，可修改项目目录权限：

chmod -R 755 /path/to/stable-diffusion-webui

或以root用户运行（不推荐，存在安全风险）。

3. 模型加载失败

确保模型文件路径正确，且在launch.py中指定--medvram或--lowvram参数（低显存用户）：

python launch.py --listen --medvram

通过以上步骤，即可在CentOS系统上成功部署Stable Diffusion系统。部署完成后，可通过Web界面输入文本提示生成图像，或通过API接口集成至其他应用。