mmartial/ComfyUI-Nvidia-Docker

解决的问题

此项目提供基于 Docker 的 ComfyUI（Stable Diffusion WebUI）部署。它解决了设置环境、管理 NVIDIA GPU 驱动程序，以及处理不同硬件世代（从 GTX 10xx 到 RTX 50xx/Blackwell）之间 Python 依赖的复杂性，同时确保主机操作系统隔离和适当的用户权限管理。

运作方式

它使用官方 NVIDIA CUDA 容器作为基础，以确保最佳的 GPU 性能。系统将安装分为两个主要目录：run 文件夹用于虚拟环境和源代码，basedir 文件夹用于用户文件、模型和自定义节点。它支持 Docker 和 Podman，并允许用户映射自己的 UID/GID，以避免容器内创建文件时出现权限问题。

适用对象

ComfyUI 的用户，想要一个可携带、隔离且易于更新，并兼容于各种 NVIDIA GPU（包括 Windows（通过 WSL2）、Linux 和 Unraid）的安装。

重点特色

• 硬件特定标签：提供多种 Ubuntu 和 CUDA 版本组合，以支持广泛的 GPU，包括针对 Blackwell（RTX 50xx）和较旧 GTX 10xx 显卡的特定标签。
• 权限管理：通过自定义 UID/GID 映射，将权限降级为普通用户，并保留主机权限。
• 弹性安装：支持 uv 以加快包安装，并包含 TORCH_LOCK 机制，以防止 PyTorch 的破坏性更新。
• 整合管理：预装 ComfyUI-Manager，方便更新和节点管理。
• 自定义：允许通过环境变量覆盖命令行参数，并在容器内执行自定义 bash 脚本。