reorganize

comfyui-rocm-dockerfile

@@ -1,14 +1,22 @@
 # Use the official ROCm base image
 FROM ubuntu:26.04 AS reqs
 
-# Set environment variables for 7900 XTX (gfx1100)
+# Set build variables
 ENV PIP_BREAK_SYSTEM_PACKAGES=1
 ENV PYTHONUNBUFFERED=1
 ENV DEBIAN_FRONTEND=noninteractive
-ENV ROCM_PATH=/opt/rocm
+
+# Set path variables
+ENV Torch_DIR=/usr/lib/python3/dist-packages/torch/share/cmake/Torch
+ENV CMAKE_PREFIX_PATH=/usr/lib/python3/dist-packages/torch/share/cmake
+ENV ROCM_PATH=/usr
+ENV HIP_PATH=/usr
+
+# Set environment variables
 ENV PYTORCH_ROCM_ARCH=gfx1100
 ENV HIP_VISIBLE_DEVICES=0
 
+# Install libraries
 RUN apt update -y && apt install -y --no-install-recommends \
     rocm \
     python3-pip \
@@ -28,6 +36,9 @@ RUN apt update -y && apt install -y --no-install-recommends \
     ninja-build \
     && rm -rf /var/lib/apt/lists/*
 
+# SymLink for stability
+RUN ln -s /usr /opt/rocm
+
 FROM reqs AS torch-audio
 
 WORKDIR /builder
@@ -35,9 +46,6 @@ WORKDIR /builder
 RUN git clone --depth 1 --branch v2.2.2 https://github.com/pytorch/audio.git && \
     cd audio && \
     git submodule update --init --recursive && \
-    export PIP_BREAK_SYSTEM_PACKAGES=1 && \
-    export Torch_DIR=/usr/lib/python3/dist-packages/torch/share/cmake/Torch && \
-    export CMAKE_PREFIX_PATH=/usr/lib/python3/dist-packages/torch/share/cmake && \
     USE_ROCM=1 BUILD_ROCM=1 CC=gcc CXX=g++ python3 setup.py install && \
     cd .. && rm -rf audio