build: Restore installation of PMI2 on Ubuntu 20.04

coreweave · Nov 1, 2024 · 06c7fa5 · 06c7fa5
1 parent 36ae5c6
commit 06c7fa5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/Dockerfile.ubuntu20 b/Dockerfile.ubuntu20
@@ -122,6 +122,11 @@ RUN cd /tmp && \
 # Rebuild OpenMPI to support SLURM
 # For Ubuntu 22, we can replace PMI2 (--with-pmi) with PMIx
 # --with-pmix=/usr/lib/x86_64-linux-gnu/pmix2
+RUN apt-get -qq update && \
+    apt-get -qq install -y --no-install-recommends \
+      libpmi2-0 libpmi2-0-dev && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
 SHELL ["/bin/bash", "-c"]
 RUN source /opt/hpcx/hpcx-init.sh && \
     hpcx_load && \