前提:
硬件环境:NVIDIA显卡4G以上、16G以上内存、至少20G硬盘空间,
软件环境:Win10以上操作系统、cuda_11.8.0_522.06_windows.exe、Github win客户端 、Miniconda3
开始:
conda create -n sd21 python=3.10.6
conda activate sd21
安装torch的GPU版本
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch(1月7日更新)
测试GPU是否驱动
python -c "import torch; print(torch.cuda.is_available())"
返回:True
下载webui
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
cd stable-diffusion-webui
新建目录repositories:
md repositories
1、git clone https://github.com/CompVis/stable-diffusion.git repositories/stable-diffusion
2、git clone https://github.com/CompVis/taming-transformers.git repositories/taming-transformers
3、git clone https://github.com/sczhou/CodeFormer.git repositories/CodeFormer
4、git clone https://github.com/salesforce/BLIP.git repositories/BLIP
5、git clone https://github.com/Stability-AI/stablediffusion.git repositories/stable-diffusion-stability-ai
6、这步删除(1月7日更新)
7、git clone https://github.com/crowsonkb/k-diffusion.git repositories/k-diffusion
安装SD环境
pip install transformers==4.19.2 diffusers invisible-watermark --prefer-binary
安装k-diffusion
pip install git+https://github.com/crowsonkb/k-diffusion.git --prefer-binary
可选 安装GFPGAN,安装后放在虚拟环境的包目录内f:\anaconda3\envs\sd20\lib\site-packages
pip install git+https://github.com/TencentARC/GFPGAN.git --prefer-binary
可选 环境安装
pip install -r repositories/CodeFormer/requirements.txt --prefer-binary
主环境安装
pip install -r requirements.txt --prefer-binary
升级numpy
pip install -U numpy --prefer-binary
安装open_clip模块
pip install open_clip_torch
拷贝模型:
v2-1_512-ema-pruned.ckpt (迅雷下载地址:https://huggingface.co/stabilityai/stable-diffusion-2-1-base/resolve/main/v2-1_512-ema-pruned.ckpt)
v2-1_512-ema-pruned.yaml (见网盘)
百度网盘:https://pan.baidu.com/s/1Nkk7_nttp8kAwzi98glwPA
提取码:j7nx
到目录stable-diffusion-webui\models\Stable-diffusion\
运行启动程序:
python webui.py --medvram (6G显存可运行)
python.exe webui.py --lowvram(4G显存可运行)
不加参数8G及以上