逍遥之韶

欲穷千里目，更上一层楼。

1028 字

5 分钟

StableDiffusion：Linux环境内安装

2024-10-22

AIGC

AIGC·图像

/

StableDiffusion

1. 基础环境安装#

显卡驱动：下载最新版本
CUDA：推荐和显卡驱动匹配的版本，下载时选本地安装模式（Local版本）
miniconda：下载较新的版本
Git：推荐最新版本

2. WebUI：官方原版#

2.1 获取仓库代码#

1
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git

2.2 创建虚拟环境#

1
conda create -n webui python=3.10
2
conda activate webui

2.3 安装pytorch#

注意pytorch版本和显卡驱动版本的兼容

1
# 最新版本
2
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
3
# 指定版本
4
pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu121

2.4 安装xformers#

注意xformers和pytorch的版本对应

1
# 最新版本
2
pip install xformers
3
# 指定版本
4
pip install xformers==0.0.28.post2 # 对应torch=2.5.0、torchvision=0.20.0、torchaudio=2.5.0

2.5 安装依赖环境#

1
# 进入项目目录
2
cd stable-diffsion-webui
3

4
# 安装依赖环境
5
pip install -r requirements_versions.txt
6
pip install -r requirements.txt

2.6 下载模型#

以SD 1.5系列的Anything-v5.0模型为例
- Anything-v5.0-PRT.safetensors的checkpoints模型文件
- animevae.pt的vae模型文件

1
# 进入项目目录
2
cd stable-diffsion-webui
3

4
# 下载checkpoints模型
5
wget -P ./models/checkpoints https://hf-mirror.com/swl-models/Anything-v5.0-PRT/resolve/main/Anything-v5.0-PRT.safetensors
6

7
# 下载vae模型
8
wget -P ./models/VAE https://hf-mirror.com/Loli-Kyn/animevae/resolve/main/animevae.pt

2.7 创建启动脚本#

创建启动脚本：start.sh
参数说明：
- xformers：启用xformers，可以极大地改善内存消耗和速度
- no-half-vae：不启用半精VAE，解决生成图片时，可能的VAE精度不足所导致的报错
- disable-safe-unpickle：不启动安全检查，不加可能导致模型加载时报错
- port：对外暴露的端口号
- theme：主题
- device-id：指定GPU

1
#!/bin/bash
2

3
export COMMANDLINE_ARGS="--xformers --no-half-vae --disable-safe-unpickle --port 18288 --theme dark --device-id 6"
4

5
python_cmd="python"
6
LAUNCH_SCRIPT="launch.py"
7

8
"${python_cmd}" "${LAUNCH_SCRIPT}" "$@"

2.8 启动#

1
bash start.sh

WebUI-Linux界面

附：无法选择vae模型？#

选择Settings标签页
在Quicksettings list中选择sd_model_checkpoints和sd_vae，并应用设置
刷新页面

附：需要安装CLIP？#

可以从clip-vit-large-patch14中获取clip资源，保存在models/openai文件夹中，文件夹需要自行新建。

附：没有中文界面？#

中文界面汉化扩展插件：https://github.com/hanamizuki-ai/stable-diffusion-webui-localization-zh_Hans

WebUI-Linux界面汉化

3. WebUI-Forge：lllyasviel整合包#

3.1 获取仓库代码#

1
git clone https://github.com/lllyasviel/stable-diffusion-webui-forge.git

3.2 创建虚拟环境#

1
conda create -n webui-forge python=3.10
2
conda activate webui-forge

3.3 安装pytorch#

注意pytorch版本和显卡驱动版本的兼容

1
# 最新版本
2
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
3
# 指定版本
4
pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu121

3.4 安装xformers#

注意xformers和pytorch的版本对应

1
# 最新版本
2
pip install xformers
3
# 指定版本
4
pip install xformers==0.0.28.post2 # 对应torch=2.5.0、torchvision=0.20.0、torchaudio=2.5.0

3.5 安装依赖环境#

1
# 进入项目目录
2
cd stable-diffsion-webui-forge
3

4
# 更新einops版本
5
sed -i 's/einops==0.4.1/einops==0.8.0/g' requirements_versions.txt
6

7
# 安装依赖环境
8
pip install -r requirements_versions.txt
9

10
# 安装缺少模块
11
pip install sacremoses

3.6 下载模型#

以SD 1.5系列的Anything-v5.0模型为例
- Anything-v5.0-PRT.safetensors的checkpoints模型文件
- animevae.pt的vae模型文件

1
# 进入项目目录
2
cd stable-diffsion-webui-forge
3

4
# 下载checkpoints模型
5
wget -P ./models/checkpoints https://hf-mirror.com/swl-models/Anything-v5.0-PRT/resolve/main/Anything-v5.0-PRT.safetensors
6

7
# 下载vae模型
8
wget -P ./models/VAE https://hf-mirror.com/Loli-Kyn/animevae/resolve/main/animevae.pt

3.7 创建启动脚本#

创建启动脚本：start.sh
参数说明：
- xformers：启用xformers，可以极大地改善内存消耗和速度
- no-half-vae：不启用半精VAE，解决生成图片时，可能的VAE精度不足所导致的报错
- disable-safe-unpickle：不启动安全检查，不加可能导致模型加载时报错
- port：对外暴露的端口号
- theme：主题
- gpu-device-id：指定GPU

1
#!/bin/bash
2

3
export COMMANDLINE_ARGS="--xformers --no-half-vae --disable-safe-unpickle --port 11888 --theme dark --gpu-device-id 6"
4

5
python_cmd="python"
6
LAUNCH_SCRIPT="launch.py"
7

8
"${python_cmd}" "${LAUNCH_SCRIPT}" "$@"

3.8 启动#

1
bash start.sh

WebUI-Forge-Linux界面

附：无法读取SD 3.5的CLIP组件？#

下载SD35_ForgeV2插件，追加并替换原有的文件夹内容：backend、modules、repositories。
设置
- UI选择：all
- checkpoint选择：SD3.5系列模型
- VAE/Text Encoder选择：clip_l.safetensors、clip_g.safetensors、t5xxl_fp8_e4m3fn.safetensors
完成上述设置后，即可正常生图

4. ComfyUI：官方原版#

4.1 获取仓库代码#

1
git clone https://github.com/comfyanonymous/ComfyUI.git

4.2 创建虚拟环境#

1
conda create -n comfyui python=3.10
2
conda activate comfyui

4.3 安装pytorch#

注意pytorch版本和显卡驱动版本的兼容

1
# 最新版本
2
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
3
# 指定版本
4
pip install torch==2.5.0 torchvision==0.20.0 torchaudio==2.5.0 --index-url https://download.pytorch.org/whl/cu121

4.4 安装依赖环境#

1
pip install -r requirements.txt

4.5 下载模型#

以SD 1.5系列的Anything-v5.0模型为例
- Anything-v5.0-PRT.safetensors的checkpoints模型文件
- animevae.pt的vae模型文件

1
# 进入项目目录
2
cd ComfyUI
3

4
# 下载checkpoints模型
5
wget -P ./models/checkpoints https://hf-mirror.com/swl-models/Anything-v5.0-PRT/resolve/main/Anything-v5.0-PRT.safetensors
6

7
# 下载vae模型
8
wget -P ./models/vae https://hf-mirror.com/Loli-Kyn/animevae/resolve/main/animevae.pt

4.6 启动#

1
python main.py --port 8188 --listen 127.0.0.1 --cuda-device 6
2
# listen指定IP地址
3
# port指定IP端口
4
# cuda-device指定GPU的ID号

ComfyUI界面

StableDiffusion：插件安装

StableDiffusion：Windows环境内安装

有朋自远方来，不亦乐乎？