Skip to content

Commit 37f92bb

Browse files
Jeff Dailypytorchmergebot
authored andcommitted
[ROCm][CI] fix nightly build after rocm 6.4 upgrade (#153253)
rocm-smi adds inclusion of drm.h and libdrm-devel package was missing Pull Request resolved: #153253 Approved by: https://github.com/jeffdaily, https://github.com/atalman Co-authored-by: Jeff Daily <jeff.daily@amd.com>
1 parent 9ae722c commit 37f92bb

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

.ci/docker/manywheel/Dockerfile_2_28

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -156,8 +156,11 @@ ENV ROCM_PATH /opt/rocm
156156
# and avoid 3.21.0 cmake+ninja issues with ninja inserting "-Wl,--no-as-needed" in LINK_FLAGS for static linker
157157
RUN python3 -m pip install --upgrade pip && \
158158
python3 -mpip install cmake==3.28.4
159+
# replace the libdrm in /opt/amdgpu with custom amdgpu.ids lookup path
159160
ADD ./common/install_rocm_drm.sh install_rocm_drm.sh
160161
RUN bash ./install_rocm_drm.sh && rm install_rocm_drm.sh
162+
# ROCm 6.4 rocm-smi depends on system drm.h header
163+
RUN yum install -y libdrm-devel
161164
ENV MKLROOT /opt/intel
162165
ADD ./common/install_rocm_magma.sh install_rocm_magma.sh
163166
RUN bash ./install_rocm_magma.sh ${ROCM_VERSION} && rm install_rocm_magma.sh

0 commit comments

Comments
 (0)