怎样让CentOS运行Stable Diffusion

AI技术
小华
2025-07-14

要在CentOS上运行Stable Diffusion,可以按照以下步骤进行:

准备工作

  1. 更新系统
sudo yum update -y
  1. 安装必要的依赖
sudo yum install -y git python3 python3-pip python3-devel gcc cmake protobuf rust
  1. 安装CUDA和cuDNN(如果你使用GPU):
  • 检查系统是否有支持CUDA编程的GPU:
cat /etc/redhat-release
lspci | grep -i nvidia
nvidia-smi
  • 安装基础依赖环境:
sudo yum install kernel-devel
  • 下载并安装NVIDIA驱动:
wget https://cn.download.nvidia.com/tesla/515.65.01/nvidia-linux-x86_64-515.65.01.run
sudo sh nvidia-linux-x86_64-515.65.01.run --kernel-source-path=/usr/src/kernels/3.10.0-1127.19.1.el7.x86_64/
  • 安装Anaconda(可选):
wget https://repo.anaconda.com/archive/Anaconda3-2024.05-Linux-x86_64.sh
bash Anaconda3-2024.05-Linux-x86_64.sh

安装Stable Diffusion

  1. 克隆Stable Diffusion仓库
git clone https://github.com/CompVis/stable-diffusion.git
cd stable-diffusion
  1. 安装依赖库
pip install -r requirements.txt
  1. 下载预训练模型(可选):
wget https://huggingface.co/CompVis/stable-diffusion-v1-5/resolve/main/dreambooth-v2.1.pt

运行Stable Diffusion

  1. 启动Web服务器
python run.py --model dreambooth-v2.1.pt --port 8000
  1. 访问Web界面

打开浏览器,访问 http://localhost:8000

使用Docker安装(可选)

  1. 安装Docker
sudo yum install -y yum-utils device-mapper-persistent-data lvm2
sudo yum-config-manager --add-repo https://download.docker.com/linux/centos/docker-ce.repo
sudo yum install docker-ce docker-ce-cli containerd.io
sudo systemctl start docker
sudo systemctl enable docker
  1. 拉取Stable Diffusion镜像
docker pull stablediffusion/stable-diffusion-webui
  1. 运行Stable Diffusion容器
docker run -it --rm -p 80:80 stablediffusion/stable-diffusion-webui
  1. 访问Web界面

打开浏览器,访问 http://localhost:80

注意事项

  • 硬件要求:Stable Diffusion对GPU有一定要求,建议使用NVIDIA GPU并安装CUDA和cuDNN。
  • 网络连接:下载预训练模型可能需要稳定的网络连接。
  • 权限问题:确保你有足够的权限来安装软件和运行Web服务器。
  • 图像尺寸问题:确保图像尺寸的长和宽都是8的倍数。
  • 模型加载失败:检查模型文件路径是否正确,确保文件未被移动或删除。
  • 生成速度慢:使用GPU加速图像生成过程,确保CUDA和其他必要的库已正确配置。
  • 内存不足:关闭不必要的应用程序,释放更多内存供Stable Diffusion使用。

通过以上步骤,你应该能够在CentOS上成功安装并运行Stable Diffusion。如果在安装过程中遇到问题,可以参考官方文档或社区论坛寻求帮助。

亿速云提供售前/售后服务

售前业务咨询

售后技术保障

400-100-2938

7*24小时售后电话

官方微信小程序