10000 bring back amdsmi · pytorch/pytorch@29d8010 · GitHub
[go: up one dir, main page]
Skip to content

Commit 29d8010

Browse files
ethanwee1ethanwee1
committed
bring back amdsmi
1 parent bc5c5a5 commit 29d8010

File tree

1 file DBDA changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

‎.ci/docker/centos-rocm/Dockerfile

+4
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,11 @@ COPY ./common/install_rocm.sh install_rocm.sh
6161
RUN bash ./install_rocm.sh
6262
RUN rm install_rocm.sh
6363
COPY ./common/install_rocm_magma.sh install_rocm_magma.sh
64+
RUN bash ./install_rocm_magma.sh ${ROCM_VERSION}
6465
RUN rm install_rocm_magma.sh
66+
COPY ./common/install_amdsmi.sh install_amdsmi.sh
67+
RUN bash ./install_amdsmi.sh
68+
RUN rm install_amdsmi.sh
6569
ENV PATH /opt/rocm/bin:$PATH
6670
ENV PATH /opt/rocm/hcc/bin:$PATH
6771
ENV PATH /opt/rocm/hip/bin:$PATH

0 commit comments

Comments
 (0)
0