[Gen AI] Stable Diffusion WebUI Docker 환경 설정 및 사용하기

💡 본 문서는 '[Gen AI] Stable Diffusion WebUI Docker로 환경 설정하기'에 대해 정리해 놓은 글입니다.
현재(24.08.03.) Stable Diffusion WebUI AUTOMATIC1111의 경우, 설치 시에 다양한 문제가 존재한다. 따라서 환경이 변경되더라도 돌아가는 도커 파일을 공유하고자 내용을 정리하였으니 참고하시기 바랍니다.

현재(24.08.03.) Stable Diffusion WebUI AUTOMATIC1111 의 경우, 설치 시에 다양한 문제가 존재한다. 패키지 디펜던시로 인해 설치하지 못하는 문제도 있겠지만, 대표적으로는 CUDA 12 버전대가 설치되어 있지 않으면, CUDA 드라이버를 업데이트하라는 메시지로 실행을 하지 못하곤 한다. 따라서 현시점에서 바로 실행 가능한 Dockerfile을 공유하고자 정리하였습니다.

(물론, 이 역시 로컬에 있는 CUDA 12버전을 활용하는 버전이기에 로컬에 CUDA 12버전 드라이버가 깔려있지 않다면 문제가 생긴다... kestr3l/stable-diffusion-webui 도커 파일을 이용하면 해결할 수 있을 것으로 보이나, 그것 역시도 문제가 생긴다.)

Stable Diffusion WebUI Docker로 환경 설정하기

siutin/stable-diffusion-webui-docker

dockerfile

# Use the base image
FROM siutin/stable-diffusion-webui-docker:latest-cuda

# Set the working directory
WORKDIR /app/stable-diffusion-webui

# Make the webui.sh script executable
RUN chmod +x webui.sh

# Expose port 7860
EXPOSE 7860

# Set the entrypoint to execute the webui.sh script
ENTRYPOINT ["bash", "webui.sh"]

# Default command for the container (can be overridden)
CMD ["--share"]

dockerfile build & execute

docker build -t my-stable-diffusion-webui .
docker run -it --name sdw --gpus all --network host -v C:/Users/PC/sjchoi/sd/webui/models:/app/stable-diffusion-webui/models -v C:/Users/PC/sjchoi/sd/webui/outputs:/app/stable-diffusion-webui/outputs -p 7860:7860 --rm my-stable-diffusion-webui

Stable Diffusion WebUI 사용하기

 

추가 자료

Docker 내부 CUDA 드라이브 설치 후 AUTOMATIC1111 실행하기

[Dockerhub] kestr3l/stable-diffusion-webui: https://hub.docker.com/r/kestr3l/stable-diffusion-webui

[Github] docker-stable-diffusion-webui: https://github.com/kestr31/docker-stable-diffusion-webui  

AUTOMATIC1111외의 여러 WebUI 사용해보기

[Dockerhub] ashleykza/stable-diffusion-webui: https://hub.docker.com/r/ashleykza/stable-diffusion-webui

[Github] Sygil-Dev/sygil-webui: https://github.com/Sygil-Dev/sygil-webui 

참고

• [Github] AUTOMATIC1111/stable-diffusion-webui: https://github.com/AUTOMATIC1111/stable-diffusion-webui
• [Dockerhub] siutin/stable-diffusion-webui-docker: https://hub.docker.com/r/siutin/stable-diffusion-webui-docker 
• [Github] siutin/stable-diffusion-webui-docker: https://github.com/siutin/stable-diffusion-webui-docker