mmartial/ComfyUI-Nvidia-Docker

해결하는 문제

이 프로젝트는 Stable Diffusion WebUI인 ComfyUI를 위한 Docker 기반 배포를 제공합니다. 환경 설정, NVIDIA GPU 드라이버 관리, 다양한 하드웨어 세대(GTX 10xx에서 RTX 50xx/Blackwell까지)에 걸친 Python 종속성 처리의 복잡성을 해결하고, 호스트 OS 분리와 적절한 사용자 권한 관리를 보장합니다.

작동 방식

공식 NVIDIA CUDA 컨테이너를 기본으로 사용하여 최적의 GPU 성능을 보장합니다. 시스템은 설치를 두 개의 주요 디렉토리로 분리합니다: run 폴더는 가상 환경과 소스 코드용, basedir 폴더는 사용자 파일, 모델, 사용자 정의 노드용입니다. Docker와 Podman을 모두 지원하며, 사용자가 자신의 UID/GID를 매핑하여 컨테이너 내에서 생성된 파일의 권한 문제를 피할 수 있습니다.

대상 사용자

휴대 가능하고 격리된, 업데이트가 쉬우며 다양한 NVIDIA GPU(Windows(WSL2 경유), Linux, Unraid 포함)와 호환되는 설치를 원하는 ComfyUI 사용자.

주요 특징

• 하드웨어별 태그: 광범위한 GPU를 지원하기 위해 여러 Ubuntu 및 CUDA 버전 조합을 제공하며, Blackwell(RTX 50xx) 및 구형 GTX 10xx 카드용 특정 태그를 포함합니다.
• 권한 관리: 사용자 정의 UID/GID 매핑을 통해 일반 사용자로 권한을 낮추고 호스트 권한을 유지합니다.
• 유연한 설치: uv를 지원하여 패키지 설치를 가속화하고, PyTorch에 대한 파괴적인 업데이트를 방지하는 TORCH_LOCK 메커니즘을 포함합니다.
• 통합 관리: ComfyUI-Manager가 사전 설치되어 업데이트와 노드 관리가 용이합니다.
• 사용자 정의: 환경 변수를 통한 명령줄 재정의와 컨테이너 내 사용자 정의 bash 스크립트 실행을 허용합니다.