Stable Diffusion在Linux上的安装指南

以下是在Linux上安装Stable Diffusion的指南：

系统要求

• 操作系统：推荐Ubuntu 20.04或更高版本。
• Python：3.8或更高版本。
• 内存：至少8GB。
• 存储：至少20GB可用空间。
• GPU：NVIDIA GPU，推荐RTX 3070或更高版本，需安装CUDA和cuDNN。

安装步骤

1. 更新系统并安装基础依赖：对于Ubuntu/Debian系统，执行sudo apt update && sudo apt upgrade -y，然后安装python3-pip、python3-venv、git、wget等。对于CentOS/RHEL系统，执行sudo yum update -y，安装相关开发工具和Python等。
2. 安装NVIDIA驱动和CUDA：根据系统类型，使用相应命令安装NVIDIA驱动和CUDA，如Ubuntu系统可执行sudo apt install -y nvidia-driver-525 cuda-11-8。
3. 配置环境变量：将CUDA和库文件路径添加到环境变量中，例如在~/.bashrc中添加export PATH=/usr/local/cuda/bin:$PATH和export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH，然后执行source ~/.bashrc。
4. 获取Stable Diffusion代码：使用git clone https://github.com/CompVis/stable-diffusion命令克隆仓库。
5. 创建虚拟环境并安装依赖项：使用python3 -m venv sd_env创建虚拟环境，然后执行source sd_env/bin/activate激活，再通过pip install -r requirements.txt安装依赖项。
6. 下载模型权重：从Hugging Face下载预训练模型权重，将其放到指定目录，如models/ldm/stable-diffusion-v1。
7. 生成图像：创建Python脚本，如stable_diffusion.py，添加代码加载模型、设置参数并生成图像，然后运行脚本。