init test

comfyui-rocm-dockerfile

@@ -1,21 +1,44 @@
 # Use the official ROCm base image
-FROM ubuntu:26.04
+FROM ubuntu:26.04 as reqs
 
 # Set environment variables for 7900 XTX (gfx1100)
 ENV PIP_BREAK_SYSTEM_PACKAGES=1
 ENV PYTHONUNBUFFERED=1
+ENV DEBIAN_FRONTEND=noninteractive
+ENV ROCM_PATH=/opt/rocm
+ENV PYTORCH_ROCM_ARCH=gfx1100
+ENV HIP_VISIBLE_DEVICES=0
+ENV PYTORCH_ROCM_ARCH=your_gfx_arch
 
 RUN apt update -y && apt install -y --no-install-recommends \
     rocm \
     python3-pip \
     python3-dev \
     python3-torch-rocm \
-    python3-torchaudio \
     git \
     libgl1 \
     libglib2.0-bin \
+    cmake \
+    ninja-build \
+    pkg-config \
+    libavcodec-dev \
+    libavformat-dev \
+    libswresample-dev \
     && rm -rf /var/lib/apt/lists/*
 
+FROM reqs as torch-audio
+
+WORKDIR /builder
+
+RUN git clone https://github.com/pytorch/audio.git && \
+    cd audio && \
+    git submodule update --init --recursive && \
+    # We must ensure it uses the system python/torch
+    BUILD_ROCM=1 python3 setup.py install && \
+    cd .. && rm -rf audio
+
+FROM torch-audio as runner
+
 WORKDIR /app
 
 # Clone ComfyUI