Linux环境下如何配置Stable Diffusion

以下是在Linux环境下配置Stable Diffusion的步骤：

1. 硬件准备：推荐NVIDIA GPU（如RTX 3060 12GB及以上）、8GB+内存、20GB+存储空间。
2. 软件准备：

• 安装Python 3.8+：sudo apt install python3（Ubuntu/Debian）或sudo yum install python3（CentOS/RHEL）。
• 创建虚拟环境：python3 -m venv sd_env，然后source sd_env/bin/activate。
• 安装依赖库：pip install torch torchvision transformers diffusers accelerate safetensors xformers。
• 安装NVIDIA驱动和CUDA（若用GPU加速）：参考NVIDIA官方文档和cuDNN官方文档进行安装和配置。

1. 获取模型：从Hugging Face Model Hub或其他可靠来源下载预训练模型，并放置到指定目录。
2. 配置运行：

• 设置环境变量（若用GPU）：export CUDA_VISIBLE_DEVICES=0。
• 运行脚本：python run.py --config config.yaml --ckpt_path your_model.pth --prompt "your prompt"（参数可根据需要调整）。