Skip to content

Commit

Permalink
and even more
Browse files Browse the repository at this point in the history
  • Loading branch information
jiridanek committed Aug 14, 2024
1 parent 98bf95b commit 913a680
Showing 1 changed file with 12 additions and 1 deletion.
13 changes: 12 additions & 1 deletion rocm/ubi9-python-3.9/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ RUN echo "[ROCm]" > /etc/yum.repos.d/rocm.repo && \
echo "enabled=1" >> /etc/yum.repos.d/amdgpu.repo && \
echo "gpgcheck=0" >> /etc/yum.repos.d/amdgpu.repo && \
yum install -y \
rocm-smi rccl && \
rocm-utils rocm-smi rccl rocm-device-libs && \
yum clean all && rm -rf /var/cache/yum
# Install more of the supporting libs
RUN yum install -y \
Expand All @@ -47,6 +47,17 @@ RUN yum install -y \
RUN yum install -y \
hipfft hipsolver miopen-hip roctracer rocrand && \
yum clean all && rm -rf /var/cache/yum

RUN yum install -y \
rocblas-devel rocrand-devel hipfft-devel miopen-hip-devel && \
yum clean all && rm -rf /var/cache/yum

RUN yum install -y \
rocm-hip-runtime \
rocm-opencl-runtime \
rocm-language-runtime \
yum clean all && rm -rf /var/cache/yum

# Install the rest of the ROCm RPMs; this way we avoid creating a single huge
# layer, that would stress OpenShift CI infrastructure too much
RUN yum install -y rocm && \
Expand Down

0 comments on commit 913a680

Please sign in to comment.