Sign inSign up

selcarpa/comfyui-rocm

By selcarpa

Updated about 13 hours ago

https://github.com/selcarpa/ComfyUI-ROCm

Image
Machine learning & AI
0

3.0K

selcarpa/comfyui-rocm repository overview

ComfyUI ROCm

License ROCm PyTorch ComfyUI

Run ComfyUI on AMD GPUs using ROCm — Pre-built Docker image for AMD Radeon RX 6000/7000/9000 series.


Quick Start

Prerequisites
  • AMD GPU: RX 6000/7000/9000 series (RDNA 2/3/4)
  • ROCm 7.2.4+ driver installed on host
  • Docker and Docker Compose installed
1. Create docker-compose.yaml
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
      - ./data/temp:/workspace/ComfyUI/temp
    environment:
      - HIP_VISIBLE_DEVICES=0
    restart: unless-stopped
2. Add User to GPU Groups
sudo usermod -a -G render,video $USER
# Log out and log back in for changes to take effect
3. Start Container
docker compose up -d

Access ComfyUI at: http://localhost:8188


Model Setup

Place model files in ./data/models/:

data/models/
├── checkpoints/    # Stable Diffusion models
├── vae/           # VAE models
├── loras/         # LoRA files
└── ...

Environment Variables

VariableDefaultDescription
HIP_VISIBLE_DEVICES0GPU device index
COMFYUI_MANAGER_DISABLEDfalseDisable auto-restore of ComfyUI-Manager

GPU Compatibility

GPU SeriesArchitectureExamples
RX 9000RDNA 4RX 9070 XT, RX 9070
RX 7000RDNA 3RX 7900 XTX, RX 7800 XT
RX 6000RDNA 2RX 6950 XT, RX 6800

Tested on: AMD Radeon RX 7800 XT (16GB VRAM, RDNA 3)


For detailed documentation, troubleshooting, and build instructions, visit:

GitHub Repository

If this image is useful, please star the repo!


License

GPL-3.0 | ComfyUI: GPL-3.0 | PyTorch: BSD 3-Clause | AMD ROCm: Open Source


中文说明

本镜像用于在 AMD GPU 上运行 ComfyUI,支持 RX 6000/7000/9000 系列显卡。

快速启动

前置要求
  • AMD GPU: RX 6000/7000/9000 系列
  • 主机已安装 ROCm 7.2.4+ 驱动
  • 已安装 Docker 和 Docker Compose
1. 创建 docker-compose.yaml
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
      - ./data/temp:/workspace/ComfyUI/temp
    environment:
      - HIP_VISIBLE_DEVICES=0
    restart: unless-stopped
2. 添加 GPU 权限
sudo usermod -a -G render,video $USER
# 注销并重新登录
3. 启动容器
docker compose up -d

访问 ComfyUI:http://localhost:8188

模型设置

将模型文件放入 ./data/models/ 目录。

相关链接

详细文档、故障排除和构建说明请访问:

GitHub 仓库

如果觉得有用,请给仓库点个 Star!


GPL-3.0 许可

Tag summary

Content type

Image

Digest

sha256:36e925088

Size

18.8 GB

Last updated

29 days ago

docker pull selcarpa/comfyui-rocm