Skip to content

Commit fe72574

Browse files
rocm-smi adds inclusion of drm.h and libdrm-devel package was missing
Ported from upstream pytorch/pytorch@37f92bb
1 parent 671588f commit fe72574

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

manywheel/Dockerfile_2_28

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,8 +154,11 @@ ENV ROCM_PATH /opt/rocm
154154
# and avoid 3.21.0 cmake+ninja issues with ninja inserting "-Wl,--no-as-needed" in LINK_FLAGS for static linker
155155
RUN python3 -m pip install --upgrade pip && \
156156
python3 -mpip install cmake==3.28.4
157+
# replace the libdrm in /opt/amdgpu with custom amdgpu.ids lookup path
157158
ADD ./common/install_rocm_drm.sh install_rocm_drm.sh
158159
RUN bash ./install_rocm_drm.sh && rm install_rocm_drm.sh
160+
# ROCm 6.4 rocm-smi depends on system drm.h header
161+
RUN yum install -y libdrm-devel
159162
ADD ./common/install_rocm_magma.sh install_rocm_magma.sh
160163
RUN bash ./install_rocm_magma.sh && rm install_rocm_magma.sh
161164
ADD ./common/install_miopen.sh install_miopen.sh

0 commit comments

Comments
 (0)