在Linux上部署Stable Diffusion需注意以下事项:
- 硬件要求:需NVIDIA GPU(推荐RTX 3070/40系,4GB+显存)、足够内存(16GB+)、存储空间(20GB+)。
- 环境配置:
- 安装NVIDIA驱动、CUDA、cuDNN,确保支持GPU加速。
- 使用Python 3.8+,创建虚拟环境并安装依赖(如
diffusers
、transformers
)。
- 网络问题:
- 国内需配置镜像源(如清华源)或代理,避免GitHub/HuggingFace访问失败。
- 模型下载需科学上网,可提前下载模型文件手动放置。
- 部署方式:
- 推荐使用Docker或直接运行WebUI脚本,避免环境冲突。
- 若用脚本部署,注意修改监听端口(
--listen
参数)和防火墙设置。
- 模型管理:
- 模型文件需放在指定目录(如
models/stable-diffusion
),支持多种格式(pt、safetensors)。 - 如需微调,可下载LoRA/Embedding模型并放置到对应目录。
- 性能优化:
- 启用混合精度(
--no-half
参数)或Xformers加速。 - 多GPU环境需配置CUDA可见性(
CUDA_VISIBLE_DEVICES
)。
- 权限与安全:
- 确保脚本有执行权限(
chmod +x webui.sh
)。 - 生产环境建议关闭调试模式,限制端口访问。
参考来源: