derek/ComfyUI-ARM64-Docker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Try to move env into the command
Docker Build and Release Upload / build-rocm (push) Failing after 3m59s
Details

Browse Source
This commit is contained in:
Derek Holloway
2026-04-27 18:38:23 -07:00
parent adb83aa4d1
commit 9093d7bbd3
1 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
comfyui-rocm-dockerfile
+9 -6
View File
@@ -7,15 +7,9 @@ ENV PYTHONUNBUFFERED=1
ENV DEBIAN_FRONTEND=noninteractive ENV DEBIAN_FRONTEND=noninteractive
# Set path variables # 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 ROCM_PATH=/usr
ENV HIP_PATH=/usr ENV HIP_PATH=/usr
# Set environment variables
ENV PYTORCH_ROCM_ARCH=gfx1100
ENV HIP_VISIBLE_DEVICES=0
# Install libraries # Install libraries
RUN apt update -y && apt install -y --no-install-recommends \ RUN apt update -y && apt install -y --no-install-recommends \
rocm \ rocm \
@@ -48,6 +42,15 @@ WORKDIR /builder
RUN git clone --depth 1 --branch v2.9.1 https://github.com/pytorch/audio.git && \ RUN git clone --depth 1 --branch v2.9.1 https://github.com/pytorch/audio.git && \
cd audio && \ cd audio && \
git submodule update --init --recursive && \ git submodule update --init --recursive && \
# Set path variables
export CMAKE_PREFIX_PATH=/usr/lib/python3/dist-packages/torch/share/cmake && \
export Torch_DIR=/usr/lib/python3/dist-packages/torch/share/cmake/Torch && \
export hip_DIR=/usr/lib/aarch64-linux-gnu/cmake/hip && \
export CMAKE_PREFIX_PATH="/usr/lib/python3/dist-packages/torch/share/cmake:/usr/lib/aarch64-linux-gnu/cmake/hip" && \
# Set environment variables
export PYTORCH_ROCM_ARCH=gfx1100 && \
export HIP_VISIBLE_DEVICES=0 && \
# Execute build
USE_ROCM=1 BUILD_ROCM=1 CC=gcc CXX=g++ python3 setup.py install && \ USE_ROCM=1 BUILD_ROCM=1 CC=gcc CXX=g++ python3 setup.py install && \
cd .. && rm -rf audio cd .. && rm -rf audio