【深度学习】AUTOMATIC1111 / stable-diffusion-webui docker-CSDN博客

代码https://github.com/AUTOMATIC1111/stable-diffusion-webui/

CUDA 11.8 制作了一个镜像可以直接开启stable diffusion的web ui 服务。

确定自己的显卡支持CUDA11.8启动此镜像方式

docker run -it --network=host  --gpus all kevinchina/deeplearning:sdwebui bash

进入容器后启动stable-diffusion-webui

su xiedong
cd /stable-diffusion-webui && ./webui.sh --server-name="0.0.0.0"

当然你可以把model文件通过docker -v挂载出来方法如下。

先启动容器

su xiedong && cd /stable-diffusion-webui && ./webui.sh --server-name="0.0.0.0"

在另外的终端将容器里的文件拷贝出来

docker cp 容器id:/stable-diffusion-webui/models ./

然后启动一个新的容器

docker run -it --network=host  --gpus all -v $PWD:/stable-diffusion-webui/models kevinchina/deeplearning:sdwebui bash

然后去容器里执行启动命令

su xiedong
cd /stable-diffusion-webui && ./webui.sh --server-name="0.0.0.0"

此时宿主机的$PWD就可以增加模型文件等操作。