ubuntu + conda 的 stable-diffusion-webui 安装笔记
前提:安装好 nvidia 显卡驱动。ubuntu 20.04, miniconda 环境
1. 下载仓库
# clone web ui and go into its directory
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui
2. 创建 conda 虚拟环境
conda create -n sd python=3.10
3. 安装所需的包
# install torch with CUDA support. See https://pytorch.org/get-started/locally/ for more instructions if this fails. 该命令会自动安装好 cuda 和 cudnn
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
安装后 测试 pytorch 是否安装好。
# -c cmd : program passed in as string (terminates option list)
python -c "import torch; print(torch.cuda.is_available())"
安装其它包:
# check if torch supports GPU; this must output "True". You need CUDA 11. installed for this. You might be able to use
# a different version, but this is what I tested.
# clone web ui and go into its directory
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui
# clone repositories for Stable Diffusion and (optionally) CodeFormer
mkdir repositories
git clone https://github.com/CompVis/stable-diffusion.git stable-diffusion-stability-ai
git clone https://github.com/CompVis/taming-transformers.git repositories/taming-transformers
git clone https://github.com/sczhou/CodeFormer.git repositories/CodeFormer
git clone https://github.com/salesforce/BLIP.git repositories/BLIP
# install requirements of Stable Diffusion
pip install transformers==4.19.2 diffusers invisible-watermark --prefer-binary
# install k-diffusion
# pip install git+https://github.com/crowsonkb/k-diffusion.git --prefer-binary
# 估计是需要替换为:
# git clone https://github.com/crowsonkb/k-diffusion.git repositories/k-diffusion
# (optional) install GFPGAN (face restoration)
pip install git+https://github.com/TencentARC/GFPGAN.git --prefer-binary
# (optional) install requirements for CodeFormer (face restoration)
pip install -r repositories/CodeFormer/requirements.txt --prefer-binary
# install requirements of web ui
pip install -r requirements.txt --prefer-binary
# update numpy to latest version
pip install -U numpy --prefer-binary
下载 sd 1.5 模型(v1-5-pruned-emaonly.ckpt)放到:stable-diffusion-webui/models/Stable-diffusion/ 下。网址:https://huggingface.co/runwayml/stable-diffusion-v1-5
问题1:Warning: k_diffusion not found at path stable-diffusion-webui/repositories/k-diffusion/k_diffusion/sampling.py
解决:
https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/1272#issuecomment-1262244055
问题2:ImportError: cannot import name 'get_device' from 'basicsr.utils.misc'
解决:https://github.com/sczhou/CodeFormer/issues/193#issuecomment-1503410609
4. 启动程序
python webui.py
上述命令会报以下问题:
问题3:No module 'xformers'. Proceeding without it.
需要使用如下命令才能使用 xformers:
# 参数列表见:modules/cmd_args.py
python webui.py --xformers
安装 xformers:
坑:不同版本的 torch 需要不同版本的 xformers。以上我们安装的是 2.0.1 版本的 torch,参考:https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/10494#issuecomment-1552353339 ,因此:
pip install xformers==0.0.20
否则你会在生成图像的时候会报如下错误:
NotImplementedError: No operator found for `memory_efficient_attention_forward` with inputs:
query : shape=(2, 4096, 8, 40) (torch.float16)
key : shape=(2, 4096, 8, 40) (torch.float16)
value : shape=(2, 4096, 8, 40) (torch.float16)
attn_bias : <class 'NoneType'>
p : 0.0
`flshattF` is not supported because:
xFormers wasn't build with CUDA support
Operator wasn't built - see `python -m xformers.info` for more info
`tritonflashattF` is not supported because:
xFormers wasn't build with CUDA support
requires A100 GPU
`cutlassF` is not supported because:
xFormers wasn't build with CUDA support
`smallkF` is not supported because:
xFormers wasn't build with CUDA support
dtype=torch.float16 (supported: {torch.float32})
max(query.shape[-1] != value.shape[-1]) > 32
unsupported embed per head: 40
最后我们就可以愉快的使用了:
python webui.py --xformers
输入 pandas:
![drawing](https://img2023.cnblogs.com/blog/1564596/202306/1564596-20230626212628935-1935932618.png)
5. 一键启动
配置一个 sd_web_ui 脚本:
#!/bin/bash
source activate sd
cd /xx/stable-diffusion-webui
python webui.py --xformers
然后我们就可以在命令行启动了:
./sd_web_ui
参考:
【1】https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Install-and-Run-on-NVidia-GPUs :Manual Installation