在CentOS上部署Stable Diffusion是一个相对复杂的过程,但通过详细的步骤和注意事项,可以确保其兼容性和稳定性。以下是一些关键步骤和注意事项:
首先,确保系统是最新的,并安装一些必要的软件包:
sudo yum update -y
sudo yum install git python3 git-devel openssl-devel zlib-devel libffi-devel cmake protobuf rust -y
安装Python 3.10.6和pip:
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$(nproc)
sudo make install
sudo ln -s /usr/local/python3/bin/python3.10 /usr/bin/python3
sudo ln -s /usr/local/python3/bin/pip3.10 /usr/bin/pip3
pip3 install --upgrade pip
要在GPU上运行Stable Diffusion,需要安装Nvidia的Container Toolkit:
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
安装Nvidia的Container Toolkit:
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/yum.repos.d/nvidia-docker.list
sudo yum install -y nvidia-docker2
sudo systemctl restart docker
下载Stable Diffusion的官方Docker镜像:
docker pull wareft/stablediffusion:1.0
运行Stable Diffusion Docker镜像:
docker run -it --rm --gpus all -p 54321:54321 wareft/stablediffusion:1.0 /bin/bash
成功运行Docker镜像后,可以在命令行下启动Jupyter Lab:
jupyter lab --ip 0.0.0.0 --port 54321 --allow-root
通过以上步骤,你应该能够在CentOS上成功安装并运行Stable Diffusion。如果在安装过程中遇到任何问题,可以参考相关的教程或社区支持。