CentOS安装Stable Diffusion内核

AI技术
小华
2025-11-03

CentOS安装Stable Diffusion内核(WebUI)详细步骤

一、前置准备

1. 硬件要求

  • GPU:必须使用NVIDIA显卡(显存≥4GB,推荐6GB及以上,如3060/3080/4090等);
  • CPU:Intel/AMD主流型号(如i5-12400、Ryzen 5 5600X);
  • 内存:≥8GB(推荐16GB及以上,避免生成大图时内存不足);
  • 存储:预留≥100GB可用空间(用于存储模型、生成的图片及临时文件)。

2. 系统更新与基础依赖安装

# 更新系统至最新版本
sudo yum update -y
# 安装基础依赖(编译工具、Python环境、网络工具等)
sudo yum install -y git python3 python3-pip python3-devel gcc cmake protobuf rust wget

3. 安装NVIDIA驱动与CUDA Toolkit(GPU加速必备)

Stable Diffusion依赖NVIDIA GPU的并行计算能力,需安装匹配的驱动和CUDA Toolkit(建议版本≥11.7):

# 添加NVIDIA官方YUM源
sudo rpm --import https://developer.download.nvidia.com/compute/cuda/repos/rhel8/x86_64/3bf863cc.pub
sudo tee /etc/yum.repos.d/nvidia-cuda.repo <

验证驱动安装

nvidia-smi  # 应显示显卡型号、驱动版本及CUDA版本(如CUDA 11.8)

4. 配置环境变量

将CUDA路径添加至系统环境变量,确保Python能调用GPU加速:

echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc

二、安装Stable Diffusion WebUI

1. 安装Python 3.10+(推荐3.10.6)

Stable Diffusion WebUI(AUTOMATIC1111版本)要求Python≥3.10,CentOS默认Python版本较低,需手动编译安装:

# 下载Python 3.10.6源码
cd /usr/local
sudo wget https://www.python.org/ftp/python/3.10.6/Python-3.10.6.tgz
sudo tar -zxvf Python-3.10.6.tgz
cd Python-3.10.6
# 编译安装(启用优化选项)
sudo ./configure --prefix=/usr/local/python3 --enable-optimizations
sudo make -j$(nproc)  # 使用多核编译(加快进度)
sudo make install
# 创建软链接(方便全局调用)
sudo ln -s /usr/local/python3/bin/python3.10 /usr/bin/python3
sudo ln -s /usr/local/python3/bin/pip3.10 /usr/bin/pip3
# 升级pip至最新版本
python3 -m pip install --upgrade pip

2. 克隆Stable Diffusion WebUI仓库

# 创建安装目录(建议使用全英文路径,无空格)
mkdir -p ~/stable-diffusion && cd ~/stable-diffusion
# 克隆AUTOMATIC1111官方WebUI仓库(最新稳定版)
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui

3. 安装Python依赖

# 创建并激活虚拟环境(隔离项目依赖,避免冲突)
python3 -m venv venv
source venv/bin/activate
# 升级pip并安装依赖
pip install --upgrade pip
pip install -r requirements.txt  # 基础依赖
pip install xformers  # 用于优化内存占用(可选但推荐)

4. 启动WebUI(一键部署)

# 运行安装脚本(自动下载模型、配置环境)
bash webui.sh
# 脚本执行流程:
# 1. 下载Stable Diffusion基础模型(如v1-5);
# 2. 安装额外的Python包;
# 3. 启动Web服务。
# 等待脚本完成(约5-10分钟,取决于网络速度)。

5. 访问WebUI

  • 脚本运行成功后,终端会显示访问地址(通常为http://127.0.0.1:7860);
  • 在浏览器中输入该地址,即可看到Stable Diffusion WebUI界面;
  • 首次使用需设置管理员账号(用于控制生成权限)。

三、常见问题解决

1. GPU无法识别

  • 原因:未正确安装CUDA/cuDNN或环境变量未配置;
  • 解决:重新检查nvidia-smi输出,确认CUDA版本与PyTorch版本匹配(如CUDA 11.8对应PyTorch 2.0+),并确保~/.bashrc中添加了CUDA路径。

2. pip安装缓慢

  • 原因:默认源在国内访问较慢;
  • 解决:更换为国内镜像源(如清华源):
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

3. 内存不足导致崩溃

  • 原因:生成大图(如1024x1024)时内存占用过高;
  • 解决
  • 关闭其他占用内存的程序;
  • 在WebUI的“Settings→Performance”中降低“Batch size”(如设置为1);
  • 使用“xformers”内存优化(已包含在依赖中)。

四、后续优化

  • 模型替换:下载第三方模型(如RealisticVision、Deliberate)替换默认模型,放置在models/Stable-diffusion目录下,重启WebUI即可切换;
  • 扩展功能:安装ControlNet插件(用于姿态控制、线稿上色),通过git clone添加至extensions目录,并在WebUI中启用;
  • 公网访问:使用Cpolar或frp等内网穿透工具,将本地WebUI映射至公网,实现远程访问(需注意网络安全,配置防火墙规则)。
亿速云提供售前/售后服务

售前业务咨询

售后技术保障

400-100-2938

7*24小时售后电话

官方微信小程序