Merge pull request #319 from nebius/fix_nccl_pmix

[Fix] Install libpmix into nccl-benchmark image
nebius · Jan 14, 2025 · a11ce28 · a11ce28
2 parents d8c09d1 + 4b85857
commit a11ce28
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/images/nccl_benchmark/nccl_benchmark.dockerfile b/images/nccl_benchmark/nccl_benchmark.dockerfile
@@ -33,7 +33,9 @@ RUN apt-get update && \
         squashfs-tools \
         zstd \
         software-properties-common \
-        vim
+        vim \
+        libpmix2 \
+        libpmix-dev
 
 # TODO: Install only necessary packages
 # Download and install Slurm packages