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