【深度学习】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就可以增加模型文件等操作。

你可能感兴趣的:(深度学习机器学习,深度学习,stable,diffusion,docker)