From fcbca5ddeb25a4c484c9f1faa05fd3fd9d9ec460 Mon Sep 17 00:00:00 2001
From: Swati Gupta <swatig@nvidia.com>
Date: Mon, 17 Mar 2025 20:43:38 +0000
Subject: [PATCH] Base nvbandwidth on cuda image

Signed-off-by: Swati Gupta <swatig@nvidia.com>
---
 deployments/container/nvbandwidth/Dockerfile | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/deployments/container/nvbandwidth/Dockerfile b/deployments/container/nvbandwidth/Dockerfile
index 12f31a1..2c70210 100644
--- a/deployments/container/nvbandwidth/Dockerfile
+++ b/deployments/container/nvbandwidth/Dockerfile
@@ -38,7 +38,15 @@ RUN git clone --branch ${NVBANDWIDTH_VERSION} --depth 1 --single-branch https://
     cmake -DMULTINODE=1 . && \
     make -j$(nproc)
 
-FROM mpioperator/openmpi:v0.6.0
+FROM nvcr.io/nvidia/cuda:12.6.2-base-ubuntu22.04
+
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    openmpi-bin \
+    openmpi-common \
+    libopenmpi-dev \
+    openssh-client \
+    && rm -rf /var/lib/apt/lists/*
 
 COPY --from=builder /bandwidthtest/nvbandwidth/nvbandwidth /usr/bin
 
+ENTRYPOINT ["mpirun"]