break down even more

opendatahub-io · Aug 15, 2024 · 697427c · 697427c
1 parent 3e67901
commit 697427c
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/rocm/ubi9-python-3.9/Dockerfile b/rocm/ubi9-python-3.9/Dockerfile
@@ -44,7 +44,15 @@ RUN yum install -y \
     yum clean all && rm -rf /var/cache/yum
 
 RUN yum install -y \
-      rocblas rocsolver rocsparse && \
+      rocblas && \
+    yum clean all && rm -rf /var/cache/yum
+
+RUN yum install -y \
+      rocsolver && \
+    yum clean all && rm -rf /var/cache/yum
+
+RUN yum install -y \
+      rocsparse && \
     yum clean all && rm -rf /var/cache/yum
 
 RUN yum install -y \
@@ -59,6 +67,11 @@ 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 \
+      hiptensor migraphx && \
+    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 && \