feat(compose): add AMD-optimized ComfyUI service

Add a new ComfyUI service to the Docker Compose configuration, specifically tailored for AMD GPUs using ROCm.

- Uses the rocm/pytorch base image for hardware acceleration
- Includes an initialization script to clone and install ComfyUI on first boot
- Configures necessary hardware device mappings (/dev/dri, /dev/kfd) and environment variables for AMD compatibility
- Sets up persistent storage and network integration with the existing proxy and AI internal networks
This commit is contained in:
2025-12-24 21:16:16 -05:00
parent 1440407c95
commit 00f4bd9e38

View File

@@ -56,6 +56,52 @@ services:
constraints: constraints:
- node.role == manager - node.role == manager
# --- IMAGE ENGINE: ComfyUI (AMD Optimized) ---
comfyui:
image: rocm/pytorch:rocm6.0_ubuntu22.04_py3.10_pytorch_2.1.2
# Command to auto-install ComfyUI on first boot
command: >
bash -c "
if [ ! -d /workspace/ComfyUI ]; then
git clone https://github.com/comfyanonymous/ComfyUI /workspace/ComfyUI;
cd /workspace/ComfyUI;
pip install -r requirements.txt;
fi;
cd /workspace/ComfyUI;
python3 main.py --listen 0.0.0.0 --port 8188 --preview-method auto
"
networks:
- ai_internal
- proxy
deploy:
replicas: 1
placement:
constraints:
- node.labels.gpu == true
# Hardware Access
devices:
- /dev/dri:/dev/dri
- /dev/kfd:/dev/kfd
cap_add:
- IPC_LOCK
- SYS_PTRACE
group_add:
- video
- render
volumes:
# --- UPDATED PATH ---
- /mnt/local-ssd/docker/comfyui:/workspace/ComfyUI
# System Hardware Maps
- /sys/class/drm:/sys/class/drm:ro
- /sys/class/kfd:/sys/class/kfd:ro
- /sys/devices:/sys/devices:ro
environment:
- HSA_OVERRIDE_GFX_VERSION=10.3.0
- CLI_ARGS=--listen 0.0.0.0 --port 8188
networks: networks:
ai_internal: ai_internal:
external: true external: true