怎样在CentOS部署Stable Diffusion

AI技术
小华
2025-05-17

在CentOS上部署Stable Diffusion涉及多个步骤,包括准备环境、安装必要的软件和库、下载和配置代码库等。以下是一个详细的教程,帮助你在CentOS系统上成功部署Stable Diffusion。

硬件和软件环境准备

  1. 内存:至少需要8GB RAM,推荐16GB或更高。
  2. GPU:必须是NVIDIA显卡,显存至少4GB,推荐6GB或8GB,更高则更好。
  3. CPU:只要显卡过关,CPU不要太差,基本可以满足需求。
  4. 存储空间:预留60G,最好100G以上,后期下载大模型会占用很大的存储空间。

安装必要的软件和库

  1. 安装Git
sudo yum install git -y
  1. 安装Python 3.10.6
cd /usr/local
wget https://www.python.org/ftp/python/3.10.6/Python-3.10.6.tgz
tar -zxvf Python-3.10.6.tgz
cd Python-3.10.6
./configure --prefix=/usr/local/python3 --with-openssl=/usr/local/openssl --with-openssl-rpathauto
make -j
make install
ln -s /usr/local/python3/bin/python3.10 /usr/bin/python3
ln -s /usr/local/python3/bin/pip3.10 /usr/bin/pip3
  1. 安装依赖库
sudo yum install zlib-devel perl-ExtUtils-MakeMaker asciidoc xmlto openssl-devel curl-devel libcurl-devel
  1. 安装Conda(可选,用于管理虚拟环境):
wget https://repo.anaconda.com/archive/Anaconda3-2022.10-Linux-x86_64.sh
sh ./Anaconda3-2022.10-Linux-x86_64.sh
echo 'export PATH="/root/anaconda3/bin:$PATH"' >> /root/.bashrc
source /root/.bashrc

下载和配置Stable Diffusion

  1. 下载Stable Diffusion代码库
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui
  1. 执行安装脚本
bash webui.sh
  1. 配置环境变量(可选):
export STABLE_DIFFUSION_HOME=/path/to/your/stable-diffusion/folder
export CUDA_VISIBLE_DEVICES=0,1,2,3

启动Stable Diffusion

  1. 进入Stable Diffusion界面
cd /path/to/stable-diffusion-webui
python3 -m venv venv
source venv/bin/activate
pip install --upgrade pip
pip install -r requirements_versions.txt
  1. 启动Stable Diffusion
python3 inference.py

使用Stable Diffusion

  1. 生成图片
python3 run.py --prompt "a girl" --output_dir ./output --seed 12345

常见问题及解决方法

  • Git克隆速度慢:设置代理或使用国内镜像。
  • pip安装慢:更换源或设置代理。
  • GPU无法使用:确保安装了与显卡相匹配的PyTorch版本,并添加--skip-torch-cuda-testCOMMANDLINE_ARGS变量。
  • 无法加载tokenizer:手动下载并放置相关文件到指定目录。

通过以上步骤,你应该能够在CentOS系统上成功部署Stable Diffusion。如果在安装过程中遇到问题,可以参考相关教程或社区资源进行排查和解决。

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

售前业务咨询

售后技术保障

400-100-2938

7*24小时售后电话

官方微信小程序