🔥 支持 AMD ROCm 的 ComfyUI — 在 AMD GPU 上运行 ComfyUI，预装优化过的 ROCm 兼容依赖。
ComfyUI 在 AMD ROCm 上运行示例
- 基础镜像:
rocm/pytorch:rocm7.2.4_ubuntu24.04_py3.12_pytorch_release_2.10.0 - Python: 3.12
- PyTorch: 2.10.0
- ROCm: 7.2.4
- ComfyUI: v0.22.0（构建时通过标签检出）
- 🎨 节点式 AI 工作流 — 可视化界面构建复杂 AI 流水线
- 🔥 AMD ROCm 优化 — 原生 AMD GPU 加速，支持 ROCm 7.2.4+
- 📦 灵活模型管理 — 手动放置模型，持久卷挂载避免重复下载
- 🧪 经过测试的兼容性 — 所有依赖均在真实 AMD 硬件上验证
- 🎯 开箱即用 — 预配置示例工作流
- 📦 ComfyUI-Manager 预装 — 内置节点管理器，无需手动安装
- 💾 持久化存储 — 模型和输出跨重启保留
docker compose up -d访问 ComfyUI: http://localhost:8188
| 组件 | 要求 |
|---|---|
| GPU | AMD RX 7000/9000+ 系列（支持 ROCm） |
| VRAM | 最低 8GB（推荐 16GB+） |
| OS | Linux（推荐 Ubuntu 24.04+） |
| Docker | 最新版本，支持 GPU |
| ROCm | 宿主机安装 7.2.4+ 驱动 |
- Ubuntu: 参见 ROCm Linux 安装指南
- Arch Linux: 参见 Arch Wiki - ROCm / AMDGPU
sudo usermod -a -G render,video $USER # 将当前用户加入 render 和 video 组- render — 访问 GPU 渲染节点
/dev/dri/render* - video — 访问 ROCm 核心设备
/dev/kfd
修改后需重新登录生效（或执行
newgrp render临时切换）。
rocminfo | grep gfx # 确认 GPU 架构（如 gfx1101）
cat /opt/rocm/.info/version # 显示 ROCm 版本
lsmod | grep amdgpu # 确认内核模块已加载docker compose up -d模型需要手动放置到挂载卷中。详见 模型设置指南。
参见 构建指南。
卷挂载已预配置，模型放入 ./data/models/ 对应子目录即可。
services:
comfyui-rocm:
image: selcarpa/comfyui-rocm:latest
container_name: comfyui-rocm
devices:
- /dev/kfd:/dev/kfd
- /dev/dri:/dev/dri
group_add:
- video
ports:
- "8188:8188"
volumes:
# 模型文件
- ./data/models:/workspace/ComfyUI/models
# 生成输出
- ./data/output:/workspace/ComfyUI/output
# 输入图片
- ./data/input:/workspace/ComfyUI/input
# 自定义节点
- ./data/custom_nodes:/workspace/ComfyUI/custom_nodes
# 用户配置
- ./data/user:/workspace/ComfyUI/user
environment:
- HIP_VISIBLE_DEVICES=0
- CUDA_VISIBLE_DEVICES=""
restart: unless-stopped运行: docker compose up -d
ComfyUI-Manager 自动恢复：镜像内置 ComfyUI-Manager，启动时若
custom_nodes卷中缺失则自动从镜像恢复，并自动添加--enable-manager参数启用管理界面。设置环境变量COMFYUI_MANAGER_DISABLED=true可关闭此行为。
- AMD Radeon RX 7800 XT（16GB VRAM）✅
- 生成时间: ~30-60s（512x512 图像）
- VRAM 占用: 基本操作 4-8GB
- 模型加载: 首次 ~30-60s，之后缓存
- 批处理: 支持多图像同时生成
- 挂载持久卷可避免重复下载模型
- 从
default模型开始，按需升级 - 使用快速 SSD 以获得最佳性能
遇到问题请提交 GitHub Issues。
- 自动模型下载 — 移除
download_models.py+models.yaml，模型改为完全手动放置（详见 MODEL_SETUP.md） - 构建时模型拷贝 — 移除 Dockerfile 中的模型相关拷贝和环境变量
- ComfyUI-Manager 集成 — 内置节点管理器，启动时自动恢复，开箱即用
- 构建系统 —
build.sh+ 构建文档，支持可配置版本和自动镜像标签 - 模型设置指南 — 明确模型放置方式
- CI/CD 自动发布 — GitHub Actions 自动构建并推送 Docker 镜像
- 健康检查 — 容器自动检测 ComfyUI 是否正常运行
| 角度 | 变更说明 |
|---|---|
| ROCm 升级 | 基础镜像从 ROCm 6.4.1 → 7.2.4，PyTorch 2.6.0 → 2.10.0，兼容更多 AMD GPU（RX 7000/9000+）；上游仓库支持 RX 6000 系列，本项目仅支持 RX 7000+ |
| ComfyUI 版本锁定 | 从拉取最新代码改为按指定标签（v0.22.0）检出，构建可复现 |
| 依赖优化 | 全面梳理 Python 依赖，剔除与 ROCm 不兼容的包，新增 comfyui-manager 等节点生态包，统一管理和版本锁定 |
| 节点管理 | 预装 ComfyUI-Manager，构建时克隆 + 备份，启动时自动恢复到 volume，无需手动安装 |
| 容器配置 | 重构 docker-compose，增加持久卷挂载（模型、输出、输入、自定义节点、配置），设置健康检查与环境变量 |
| 构建系统 | build.sh 支持通过环境变量配置 ComfyUI/Manager/基础镜像版本，自动生成带时间戳的详细标签 |
本项目采用 GPL-3.0 许可。详见 LICENSE 文件。
鸣谢:
- ComfyUI — 节点式 AI 工作流界面
- ComfyUI-Manager — 自定义节点管理器
- ComfyUI-ROCm — 上游 AMD ROCm 移植参考仓库
- AMD ROCm — 开源 GPU 计算平台
- ROCm 社区对 AMD GPU AI 的支持
🔗 链接: Docker Hub | GitHub | ComfyUI