56 lines
2.0 KiB
Docker
56 lines
2.0 KiB
Docker
# ComfyUI on ROCm for the AMD Strix Halo APU (gfx1151).
|
|
#
|
|
# The official `rocm/comfyui` and pytorch.org ROCm wheels compile kernels for
|
|
# datacenter archs (gfx942/gfx950) and omit gfx1151. This base — AMD's
|
|
# gfx1151-tagged PyTorch build (torch 2.9.1+rocm7.12.0rc1) — is the verified
|
|
# working GPU path on Dankclaw (2026-08-14: avail=True, Radeon 8060S, cap (11,5)).
|
|
#
|
|
# The base is heavier than a dedicated ComfyUI image (it carries vLLM), but it
|
|
# is the only prebuilt image that ships gfx1151 torch kernels. See
|
|
# docs/Research/ComfyUI ROCm and Vulkan.md.
|
|
|
|
FROM docker.io/rocm/vllm:rocm7.12.0_gfx1151_ubuntu24.04_py3.12_pytorch_2.9.1_vllm_0.16.0
|
|
|
|
# ComfyUI source + runtime requirements (non-GPU deps install from PyPI).
|
|
ENV PYTHONUNBUFFERED=1 \
|
|
COMFYUI_DIR=/opt/ComfyUI
|
|
|
|
WORKDIR /opt
|
|
|
|
# git ships in the base; clone the stable ComfyUI tag.
|
|
RUN git clone --depth 1 --branch v0.3.6 \
|
|
https://github.com/Comfy-Org/ComfyUI.git "$COMFYUI_DIR"
|
|
|
|
WORKDIR "$COMFYUI_DIR"
|
|
|
|
# Install ComfyUI Python requirements. Do NOT reinstall torch/torchvision —
|
|
# keep the gfx1151-enabled torch from the base. The requirements pin torch,
|
|
# so install everything else from a deps-only pass.
|
|
RUN /opt/python/bin/pip install --no-cache-dir \
|
|
-r requirements.txt \
|
|
--no-deps 2>/dev/null || true
|
|
|
|
# Ensure the managed deps that are missing get installed without touching torch.
|
|
# (First pass is best-effort; a reinstall of torch is deliberately avoided.)
|
|
RUN /opt/python/bin/python -m pip install --no-cache-dir \
|
|
requests==2.32.3 \
|
|
aiohttp==3.10.10 \
|
|
yarl==1.13.1 \
|
|
pillow==10.4.0 \
|
|
numpy>=1.24 \
|
|
huggingface-hub>=0.24 \
|
|
einops \
|
|
comfyui-frontend-package>=1.0 \
|
|
model-management>=1.0.3 \
|
|
opaque>=0.0.10 \
|
|
pydantic>=2.6 \
|
|
websockets>=15 \
|
|
imageio
|
|
|
|
# Run ComfyUI as the host user so bind mounts /opt/comfyui are writable.
|
|
USER 1000:1000
|
|
|
|
EXPOSE 8188
|
|
|
|
ENTRYPOINT ["/opt/python/bin/python", "main.py", "--listen", "0.0.0.0", "--port", "8188"]
|