docker部署stable-diffusion-webui

原创已于 2024-04-19 17:30:24 修改 · 1.8k 阅读

17 ·

本内容遵循CC 4.0 BY-SA版权协议

GEO检测

标签

#docker #stable diffusion #容器

于 2024-04-19 17:29:37 首次发布

本文记录在Ubuntu 20.4上通过Docker部署stable-diffusion-webui的过程，包括配置nvidia-docker2仓库，安装nvidia-docker2，验证CUDA镜像，拉取和运行特定镜像，以及解决运行sdxl模型时的tokenizer问题。

大模型弄好了，想着玩玩stable-diffusion-webui，结果折腾了几天无果，最后使用大佬的docker镜像弄好了。

Ubuntu20.4 docker运行stable diffusion webui_siutin/stable-diffusion-webui-docker-CSDN博客

以下作为备忘：

配置nvidia-docker2仓库：

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)

curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt-get update

安装nvidia-docker2并重启docker服务

sudo apt-get install -y nvidia-docker2

sudo pkill -SIGHUP dockerd

sudo systemctl daemon-reload
sudo systemctl restart docker

运行cuda镜像验证是否安装成功

sudo docker run --rm --gpus all nvidia/cuda:11.6.2-base-ubuntu20.04 nvidia-smi

拉取镜像

docker pull siutin/stable-diffusion-webui-docker:latest-cuda

支持sdxl

运行sdxl模型时会提示 Can't load tokenizer for 'openai/clip-vit-large-patch14'

这是缺少openai的工具包，手工下载映射一下即可

git clone https://www.modelscope.cn/AI-ModelScope/clip-vit-large-patch14.git

然后运行镜像

docker run -it -d --name zhangpeng -p 7860:7860 --gpus all --network host -v /data/stable-diffusion-webui/openai:/app/stable-diffusion-webui/openai -v /data/stable-diffusion-webui/models:/app/stable-diffusion-webui/models -v /data/stable-diffusion-webui/outputs:/app/stable-diffusion-webui/outputs --rm xxx.tencentcloudcr.com/xxxx/stable-diffusion-webui-docker:latest-cuda bash webui.sh --share --listen --gradio-auth zhangpeng:abc@1234