Fix version mismatch
Docker Build and Release Upload / build-rocm (push) Failing after 20s

This commit is contained in:
2026-04-27 18:22:47 -07:00
parent 02b05a99f1
commit eee8462216
+4 -1
View File
@@ -22,6 +22,9 @@ RUN apt update -y && apt install -y --no-install-recommends \
python3-pip \ python3-pip \
python3-dev \ python3-dev \
python3-torch-rocm \ python3-torch-rocm \
libhipblas-dev \
librocblas-dev \
hip-dev \
git \ git \
libgl1 \ libgl1 \
libglib2.0-bin \ libglib2.0-bin \
@@ -43,7 +46,7 @@ FROM reqs AS torch-audio
WORKDIR /builder WORKDIR /builder
RUN git clone --depth 1 --branch v2.2.2 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 && \
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 && \