docker部署stable-diffusion-webui_stable diffusion 最新镜像版本部署

大模型弄好了，想着玩玩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