如何在Linux上运行Stable Diffusion模型

以下是在Linux上运行Stable Diffusion模型的步骤：

1. 环境准备

• 系统：推荐Ubuntu 20.04+，需安装NVIDIA GPU驱动（支持CUDA）。
• 硬件：至少8GB GPU显存、16GB内存、20GB存储空间。

1. 安装依赖

• 创建虚拟环境：python3 -m venv sd_env，激活：source sd_env/bin/activate。
• 安装基础库：pip install torch torchvision transformers diffusers accelerate safetensors。
• 安装CUDA工具包（若需GPU加速）：参考NVIDIA官方文档，确保与驱动版本匹配。

1. 下载模型与代码

• 从Hugging Face获取模型权重（如CompVis/stable-diffusion-v1-4）。
• 克隆代码仓库（如AUTOMATIC1111/stable-diffusion-webui）。

1. 配置运行环境

• 设置CUDA可见设备：export CUDA_VISIBLE_DEVICES=0（指定GPU编号）。
• 运行脚本：python run.py --prompt "your text" --ckpt_path /path/to/model.ckpt（参数根据项目调整）。

1. 生成图像

• 指定输出路径和参数（如分辨率、步数），生成图像保存至指定目录。

参考来源：