diff --git a/Dockerfile.ubuntu20 b/Dockerfile.ubuntu20
index 44c7ed8..c661d51 100644
--- 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 && \