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