Skip to content

Commit 5e43f90

Browse files
Add libibumad3 and mlnx-tools packages to KFTO CUDA image
1 parent f65740b commit 5e43f90

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

images/runtime/training/cuda/Dockerfile

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,9 +57,11 @@ RUN dnf config-manager \
5757
&& dnf install -y \
5858
libibverbs-utils \
5959
infiniband-diags \
60-
libibumad \
60+
libibumad3 \
6161
librdmacm \
62+
librdmacm-utils \
6263
rdma-core \
64+
mlnx-tools \
6365
&& dnf clean all \
6466
&& rm -rf /var/cache/dnf/*
6567

0 commit comments

Comments
 (0)