Fix paths

2026-04-27 18:06:23 -07:00
parent af56bb6603
commit 917b15f35e
1 changed files with 2 additions and 0 deletions
@@ -36,6 +36,8 @@ 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