当前位置:AIGC资讯 > AIGC > 正文

Stable Diffusion 安装

一、安装 GPU 驱动(CPU 环境跳过这一步)

手动安装GPU驱动参考:https://docs.nvidia.com/datacenter/tesla/tesla-installation-notes/index.html?spm=a2c4g.11186623.0.0.2d45639cNEukDR

下载链接:https://www.nvidia.com/Download/Find.aspx?spm=a2c4g.11186623.0.0.503f573bVfJPKT&lang=cn

二、配置 pip 国内源

创建配置文件

vi ~/.pip/pip.conf

写入配置

[global]
index-url=http://mirrors.cloud.aliyuncs.com/pypi/simple/
 
[install]
trusted-host=mirrors.cloud.aliyuncs.com

三、安装 python 3.10.6(建议使用3.10版本)

这里使用的是 ubantu 安装

3.1.1 安装相关依赖

安装依赖

apt install build-essential zlib1g-dev libncurses5-dev libgdbm-dev libnss3-dev libssl-dev libreadline-dev libffi-dev libbz2-dev liblzma-dev libgdbm-compat-dev
3.1.2 安装 openssl (不安装后面运行会各种报错)

下载

wget http://www.openssl.org/source/openssl-1.1.1.tar.gz

解压

tar xf openssl-1.1.1.tar.gz

进入目录

cd openssl-1.1.1/
3.1.3 配置环境变量

创建自定义环境变量文件

vim .bashrc

写入环境变量

export LD_LIBRARY_PATH=/usr/local/openssl/lib:/usr/local/cuda-11.4/lib64:$LD_LIBRARY_PATH
3.1.4 安装 python

下载

wget  https://www.python.org/ftp/python/3.10.6/Python-3.10.6.tgr

解压

tar -xf Python-3.10.6.tar

进入目录

cd Python-3.10.6

使用 configure 对文件进行配置,并使用 --prefix=/usr/local/python3 指定安装路径,启用 SSL

./configure --prefix=/usr/local/python3 --with-ssl=/usr/local/openssl  #启用SSL,否则后面运行时会报错

编译安装

make && make install

软链接

ln -s /usr/local/python3/bin/python3.10 /usr/bin/python
ln -s /usr/local/python3/bin/pip /usr/bin/pip
ln -s /usr/local/python3/bin/pip3 /usr/bin/pip3

配置自定义环境变量文件

vim ~/.bashrc

写入配置

export PATH=/usr/local/python3/bin:/usr/local/cuda-11.4/bin:$PATH

加载环境变量

. ~/.bashrc

升级 pip

python3 -m pip install --upgrade pip

四、安装 Git

安装

apt-get install git

五、安装依赖

pip3 install transformers==4.24.0
pip3 install diffusers==0.7.2
pip3 install cython==0.29.33 
pip3 install fastapi==0.90.1

六、下载项目

拉取项目到本地

git clone https://github.com/IDEA-CCNL/stable-diffusion-webui.git

进入项目目录

cd stable-diffusion-webui/

七、安装项目对应 python 依赖包

GPU环境的用户需要安装与cuda版本对应的torch,cpu环境则没有具体要求

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 --extra-index-url https://download.pytorch.org/whl/cu117

pip 换源

pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/ 
# 清华大学 :https://pypi.tuna.tsinghua.edu.cn/simple/
# 中国科学技术大学 :http://pypi.mirrors.ustc.edu.cn/simple/
# 华中科技大学:http://pypi.hustunique.com/
# 豆瓣源:http://pypi.douban.com/simple/
# 腾讯源:http://mirrors.cloud.tencent.com/pypi/simple
# 华为镜像源:https://repo.huaweicloud.com/repository/pypi/simple/

继续安装依赖

pip install -r requirements_versions.txt

八、下载模型文件

进入模型存放目录

cd models/Stable-diffusion/

下载模型

wget https://huggingface.co/CompVis/stable-diffusion-v-1-4-original/resolve/main/sd-v1-4-full-ema.ckpt

九、启动项目

回到项目根目录

cd /root/stable-diffusion-webui/

CPU 环境提前执行以下命令

export COMMANDLINE_ARGS=--skip-torch-cuda-test

执行启动文件

python3 launch.py --port 1234 --listen --enable-insecure-extension-access

更新时间 2023-11-23