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

代码：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就可以增加模型文件等操作。