fix branch version

2026-04-27 17:52:53 -07:00
parent a182452605
commit af56bb6603
1 changed files with 2 additions and 2 deletions
@@ -32,11 +32,11 @@ FROM reqs AS torch-audio

 WORKDIR /builder

-RUN git clone --depth 1 https://github.com/pytorch/audio.git && \
+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 && \
-    CC=gcc CXX=g++ BUILD_ROCM=1 python3 setup.py install && \
+    USE_ROCM=1 BUILD_ROCM=1 CC=gcc CXX=g++ python3 setup.py install && \
    cd .. && rm -rf audio

 FROM torch-audio AS runner