技术分享

怎么绘事呢!——本地部署SD WebUI小记

————梦开始的地方————

本地部署stable-diffusion-webui——以Ubuntu22.04为例

系统:Ubuntu22.04

显卡:GTX1660s

stable-diffusion版本:2.1

首先要把sd2.1克隆到本地,路径默认为~/stable-diffusion-webui,也就是你的用户主目录。若要想更改,可以等安装完成后手动移动。

cd ~
git lfs clone https://github.com/AUTOMATIC1111/stable-diffusion-webui.git

第二步,运行webui.sh,将会自动配置venv并下载需要的文件。

bash webui.sh

如果在运行webui.py时,pip下载等出现问题,可以手动下载(一般创建venv不会出现问题)

进入虚拟环境:

source /home/username/stable-diffusion-webui/venv/bin/activate

这里使用中科大的源,安装./requirements_versions.txt中提到的库

注意,应当先安装torch再安装BasicSR。BasicSR是基于pytorch的包,不安装pytorch直接安装BasicSR会报错

pip install -r ./requirements_versions.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

vnev搭建完成后,重新运行webui.sh,将会自动下载所需文件。在途中遇到问题,亦可退出手动安装。如自动安装时卡在安装open-clip,可退出并使用pip安装。

pip install open-clip-torch

完成后,将模型文件放置于~/stable-diffusion-webui/models/Stable-diffusion,运行:

python webui.py

如果提示不能使用GPU,可以去Software&Updates中下载安装英伟达的官方驱动。

如果成功,可以继续安装以下组件(经测试,xformers==0.0.14无法正常安装。读者可以自行测试)

pip install xformers==0.0.16dev432
pip install triton

以后的启动可加上–xformers的参数

python webui.py --xformers

针对我的显卡(1660s),使用如下参数启动

python webui.py --medvram --deepdanbooru --xformers --precision full --no-half

至此,stable-diffusion-webui部署完成。

文章结构较乱,安装时应当适当调整!

Leave a Reply

Your email address will not be published.Required fields are marked *