This commit is contained in:
@@ -36,6 +36,8 @@ 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 && \
|
||||||
|
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 && \
|
USE_ROCM=1 BUILD_ROCM=1 CC=gcc CXX=g++ python3 setup.py install && \
|
||||||
cd .. && rm -rf audio
|
cd .. && rm -rf audio
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user