This commit is contained in:
@@ -22,6 +22,9 @@ RUN apt update -y && apt install -y --no-install-recommends \
|
||||
python3-pip \
|
||||
python3-dev \
|
||||
python3-torch-rocm \
|
||||
libhipblas-dev \
|
||||
librocblas-dev \
|
||||
hip-dev \
|
||||
git \
|
||||
libgl1 \
|
||||
libglib2.0-bin \
|
||||
@@ -43,7 +46,7 @@ FROM reqs AS torch-audio
|
||||
|
||||
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 && \
|
||||
git submodule update --init --recursive && \
|
||||
USE_ROCM=1 BUILD_ROCM=1 CC=gcc CXX=g++ python3 setup.py install && \
|
||||
|
||||
Reference in New Issue
Block a user