CentOS安装Stable Diffusion成功

在CentOS上安装Stable Diffusion模型需要一些步骤，包括安装Python环境、依赖库以及模型本身。以下是一个基本的指南：

前提条件

1. 操作系统：CentOS 7或8。
2. Python版本：建议使用Python 3.8或更高版本。
3. GPU支持（可选但推荐）：NVIDIA GPU和CUDA。

安装步骤

1. 更新系统

sudo yum update -y

2. 安装Python和pip

sudo yum install -y python3 python3-pip

3. 安装CUDA和cuDNN（如果使用GPU）

• CUDA Toolkit：

下载并安装适合你GPU的CUDA Toolkit：CUDA Toolkit Archive

sudo yum localinstall /path/to/cuda-repo-rhel7-.rpm
sudo yum install -y cuda

• cuDNN：

下载cuDNN库并解压到CUDA目录：

tar -xzvf cudnn--linux-x64-v.tgz
sudo cp cuda/include/cudnn*.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

4. 安装PyTorch

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

注意：根据你的CUDA版本选择合适的PyTorch安装命令。

5. 安装Stable Diffusion依赖库

pip3 install transformers datasets opencv-python-headless

6. 克隆Stable Diffusion仓库

git clone https://github.com/lucidrains/stable-diffusion.git
cd stable-diffusion

7. 安装Stable Diffusion模型

你可以从Hugging Face或其他来源下载预训练模型。以下是一个示例：

pip3 install git+https://github.com/lucidrains/stable-diffusion.git@main

8. 运行Stable Diffusion

进入模型目录并运行示例代码：

cd models/ldm
python3 inference.py --model_name "your-model-name" --prompt "your-prompt"

注意事项

• GPU支持：如果你没有GPU，可以尝试使用CPU版本的Stable Diffusion，但速度会慢很多。
• 内存要求：Stable Diffusion模型对内存要求较高，确保你的系统有足够的内存。
• 网络连接：下载依赖库和模型可能需要稳定的网络连接。

通过以上步骤，你应该能够在CentOS上成功安装并运行Stable Diffusion模型。如果有任何问题，请参考相关文档或社区支持。