怎么优化CentOS部署Stable Diffusion

在 CentOS 上部署和优化 Stable Diffusion，核心目标是：显卡能用、显存不爆、推理更快、环境稳定。下面按「系统 → 驱动 → 环境 → 推理优化 → 常见问题」给你一套实战优化方案。

一、系统层面优化（CentOS）

1️⃣ 系统版本建议

• ✅ CentOS 7 / Rocky / Alma 8+
• 不建议 CentOS 6（内核太老，CUDA 支持差）

# 查看系统
cat /etc/centos-release
uname -r

2️⃣ 关闭 SELinux & 防火墙（调试阶段）

setenforce 0
sed -i 's/^SELINUX=.*/SELINUX=disabled/' /etc/selinux/config
systemctl stop firewalld
systemctl disable firewalld

3️⃣ 内核参数优化（显存 & 进程）

cat >> /etc/sysctl.conf << EOF
vm.swappiness=10
fs.file-max=1000000
EOF
sysctl -p

二、显卡 & 驱动优化（最关键）

1️⃣ 确认显卡

lspci | grep -i nvidia

2️⃣ 安装 NVIDIA 驱动（推荐方式）

✅ 使用官方 run 文件（最稳定）

# 安装依赖
yum install -y gcc kernel-devel kernel-headers
# 禁用 nouveau
echo "blacklist nouveau" > /etc/modprobe.d/blacklist-nouveau.conf
echo "options nouveau modeset=0" >> /etc/modprobe.d/blacklist-nouveau.conf
# 重建 initramfs
dracut --force
reboot

安装驱动：

chmod +x NVIDIA-Linux-x86_64-*.run
./NVIDIA-Linux-x86_64-*.run

验证：

nvidia-smi

3️⃣ 安装 CUDA & cuDNN

推荐组合：

• CUDA 11.8
• cuDNN 8.7+

# 示例
export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH

三、Python & 环境优化

1️⃣ 使用 Miniconda（强烈推荐）

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

创建独立环境：

conda create -n sd python=3.10 -y
conda activate sd

2️⃣ PyTorch 安装（GPU）

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

验证：

import torch
print(torch.cuda.is_available())

四、Stable Diffusion 部署优化

1️⃣ 推荐版本

• ✅ Stable Diffusion WebUI (AUTOMATIC1111)
• ✅ ComfyUI（更省显存）

2️⃣ WebUI 启动参数优化（非常重要）

export COMMANDLINE_ARGS="--medvram --xformers --no-half-vae"

解释：

3️⃣ xformers 编译（CentOS 常见坑）

pip install xformers --index-url https://download.pytorch.org/whl/cu118

或源码（慢但稳定）：

pip install -U git+https://github.com/facebookresearch/xformers.git

五、显存 & 推理速度优化

✅ 1️⃣ 模型选择

✅ 2️⃣ 推理参数优化

• 降低分辨率：512x512
• 减少采样步数：20~30
• 使用 DPM++ 2M Karras

✅ 3️⃣ 使用 FP16 / FP8

--precision full --no-half

或：

--medvram --xformers

六、Web 访问 & 安全

1️⃣ 开放远程访问

export COMMANDLINE_ARGS="--listen --port 7860"

防火墙放行：

firewall-cmd --add-port=7860/tcp --permanent
firewall-cmd --reload

2️⃣ 使用 Nginx 反向代理（可选）

location / {
proxy_pass http://127.0.0.1:7860;
}

七、常见问题 & 解决方案

❌ 1. GLIBC / GCC 版本过低

yum install -y devtoolset-11
scl enable devtoolset-11 bash

❌ 2. 黑屏 / 无输出

nvidia-smi
journalctl -xe

❌ 3. 显存爆了

• 加 --medvram
• 用 SD 1.5
• 降低 batch size

八、推荐部署组合（总结）

✅ 最佳实践

• OS：Rocky / Alma 8
• GPU：RTX 3060 及以上
• Python：3.10
• PyTorch：CUDA 11.8
• WebUI：AUTOMATIC1111
• 启动参数：--medvram --xformers

如果你愿意，我可以帮你：
✅ 写 一键部署脚本
✅ 针对你的 显卡型号精准优化配置
✅ 给你 ComfyUI vs WebUI 对比方案
你现在的 显卡型号 + CentOS 版本 是多少？