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

XD7429716362023-11-01 12:31

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

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

shell 复制代码

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

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

shell 复制代码

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

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

先启动容器：

shell 复制代码

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

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

shell 复制代码

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

然后启动一个新的容器：

shell 复制代码

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

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

shell 复制代码

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

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