云计算、AI、云原生、大数据等一站式技术学习平台

网站首页 > 教程文章 正文

保姆级教程:基于docker镜像方式部署stable diffusion

jxf315 2024-12-15 14:02:46 教程文章 45 ℃

1、下载基础环境镜像

docker pull registry.cn-hangzhou.aliyuncs.com/modelscope-repo/modelscope:ubuntu20.04-cuda11.7.1-py38-torch2.0.1-tf1.15.5-1.8.0

2、运行基础环境镜像

docker run -itd --restart=always --name stable-diffusion-web -gpus all -p 6666:6666 --entrypoint='bash' -v /data:/opt registry.cn-hangzhou.aliyuncs.com/modelscope-repo/modelscope:ubuntu20.04-cuda11.7.1-py38-torch2.0.1-tf1.15.5-1.8.0

其中CUDA_VISIBLE_DEVICES里的序号表示使用宿主机上的哪些GPU

3、进入容器内部

docker exec -it stable-diffusion-web bash

4、拉取stable diffusion web源码

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

进入源码目录后,切换到v1.5.2-RC分支

cd stable-diffusion-webui
git checkout v1.5.2-RC

5、修改webui.sh脚本,将里面的 can_run_as_root=0改为can_run_as_root=1

6、执行脚本

nohup ./webui.sh --server-name 0.0.0.0 --port 6666 --enable-insecure-extension-access --xformers --no-half-vae --theme dark > app.log 2>&1 &

7、下载vae插件,让色彩饱和度更高

https://huggingface.co/stabilityai/sd-vae-ft-mse-original/tree/main

放置到models/VAE目录下,然后在web ui里的settings选项卡里的Stable Diffusion里的SD VAE选项里可以看到相关的插件。

最近发表
标签列表