This commit is contained in:
@@ -32,11 +32,11 @@ FROM reqs AS torch-audio
|
|||||||
|
|
||||||
WORKDIR /builder
|
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 && \
|
cd audio && \
|
||||||
git submodule update --init --recursive && \
|
git submodule update --init --recursive && \
|
||||||
export PIP_BREAK_SYSTEM_PACKAGES=1 && \
|
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
|
cd .. && rm -rf audio
|
||||||
|
|
||||||
FROM torch-audio AS runner
|
FROM torch-audio AS runner
|
||||||
|
|||||||
Reference in New Issue
Block a user