Frank
354ddc473f
Replace the CPU stock image with a container built from Containerfile.comfyui on AMD's gfx1151-tagged vLLM base — the only torch build verified to report torch.cuda.is_available()=True on the Strix Halo APU (Radeon 8060S, capability (11,5)). - Containerfile bakes the ComfyUI code in and installs its deps, but skips the torch family (the base image already ships the gfx1151 ROCm torch/torchvision/torchaudio; an unpinned 'torch' in requirements.txt would clobber the gfx1151 kernels). - compose comfyui service: build + image comfyui-rocm:gfx1151, passes /dev/kfd + /dev/dri, adds video+render groups, drops --cpu. Models/data persist under /opt/comfyui. - Un-skip the four compose-wiring acceptance tests (the gate for this work).
25 lines
1.3 KiB
Docker
25 lines
1.3 KiB
Docker
# ComfyUI on the gfx1151 (AMD Strix Halo / Radeon 8060S) ROCm torch build.
|
|
#
|
|
# Base image: AMD's gfx1151-tagged vLLM image. It is the only verified build
|
|
# where torch.cuda.is_available() is True on this APU (device capability
|
|
# (11,5) = gfx1151). It already ships the matching ROCm torch family
|
|
# (torch 2.9.1+rocm, torchvision 0.24.0+rocm, torchaudio 2.9.0+rocm), so we
|
|
# must NOT let pip reinstall them: ComfyUI's requirements.txt lists an unpinned
|
|
# `torch`, which would otherwise clobber the gfx1151 kernels with a default build.
|
|
FROM docker.io/rocm/vllm:rocm7.12.0_gfx1151_ubuntu24.04_py3.12_pytorch_2.9.1_vllm_0.16.0
|
|
|
|
# Bake the ComfyUI code into the image (rebuild the image to update the code).
|
|
RUN git clone --depth 1 https://github.com/comfyanonymous/ComfyUI /opt/ComfyUI
|
|
WORKDIR /opt/ComfyUI
|
|
|
|
# Pre-create the user-data dirs. The host mounts its collections over them
|
|
# (see the comfyui service in compose.yaml).
|
|
RUN mkdir -p /opt/ComfyUI/models /opt/ComfyUI/input /opt/ComfyUI/output /opt/ComfyUI/custom_nodes
|
|
|
|
# Install ComfyUI's dependencies, but skip the torch family (already present
|
|
# above as the gfx1151 ROCm builds). torchsde and the rest are kept.
|
|
RUN grep -viE '^(torch|torchvision|torchaudio)([^a-z]|$)' requirements.txt > /tmp/req-notorch.txt \
|
|
&& pip install --no-cache-dir -r /tmp/req-notorch.txt
|
|
|
|
EXPOSE 8188
|