mmartial/ComfyUI-Nvidia-Docker

해결하는 문제

이 프로젝트는 NVIDIA GPU에 최적화된 Docker 기반 환경을 제공하여 ComfyUI(Stable Diffusion WebUI)를 실행합니다. 복잡한 수동 설치, 호스트 OS 오염, 종속성 충돌 문제를 해결하기 위해 CUDA, CuDNN, Python 가상 환경을 포함한 전체 설정을 컨테이너 내부에 봉인합니다.

작동 방식

이 프로젝트는 공식 NVIDIA CUDA 컨테이너를 기반으로 합니다. 설치를 두 가지 주요 디렉터리로 분리합니다: run 폴더(가상 환경과 ComfyUI 소스 코드 포함)와 basedir 폴더(사용자 파일, 모델, 사용자 정의 노드 포함). 이 아키텍처는 환경 관리를 독립적으로 하면서도 사용자 데이터를 보존할 수 있게 합니다. 다양한 Ubuntu 및 CUDA 버전 조합을 지원하여, 오래된 GTX 10xx 시리즈부터 최신 Blackwell(RTX 50xx) GPU까지 다양한 NVIDIA 하드웨어와 호환되도록 합니다. 또한 ComfyUI-Manager를 통합하여 업데이트를 쉽게 하고, 사용자가 자신의 UID/GID를 매핑하여 호스트에서 파일 권한을 적절히 유지할 수 있도록 합니다.

대상 사용자

Docker 또는 Podman을 사용하여 NVIDIA GPU에서 ComfyUI를 실행하고 싶은 사용자, 특히 WSL2를 통해 Windows, Linux, Unraid 사용자로서 프로세스 격리와 간편한 설치 과정을 선호하는 사용자.

주요 특징

• 광범위한 하드웨어 지원: RTX 50xx(Blackwell) 및 GTX 10xx GPU에 특화된 다양한 CUDA 버전용 전용 컨테이너 태그 제공.
• 권한 관리: 컨테이너 사용자가 호스트 사용자 권한과 일치하도록 사용자 정의 UID/GID 매핑 지원.
• 유연한 설치: uv를 통한 더 빠른 Python 패키지 설치 및 TORCH_LOCK를 통한 업데이트로 인한 손상 방지 옵션 지원.
• 사용자 정의 가능성: 사용자 정의 bash 스크립트(user_script.bash) 실행 및 환경 변수를 통한 추가 명령줄 인수 전달 가능.
• 사전 빌드된 이미지: DockerHub에 사용 가능한 사전 빌드된 이미지 제공, Unraid 호환 버전 포함.