From 7501d81d7e0cc6919e51aa5bbe87b1a8614015a7 Mon Sep 17 00:00:00 2001 From: GitHub Date: Sun, 19 Oct 2025 00:01:22 +0000 Subject: [PATCH] chore: update nvidia test dependencies --- test/images/nvidia-training/Dockerfile | 4 ++-- test/images/nvidia/Dockerfile | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/test/images/nvidia-training/Dockerfile b/test/images/nvidia-training/Dockerfile index 053a0970c..018677c39 100644 --- a/test/images/nvidia-training/Dockerfile +++ b/test/images/nvidia-training/Dockerfile @@ -111,7 +111,7 @@ RUN curl -sL https://efa-installer.amazonaws.com/aws-efa-installer-$EFA_INSTALLE && cd && rm -rf /tmp/aws-efa-installer # Install NCCL -ARG LIBNCCL_VERSION=2.27.7-1 +ARG LIBNCCL_VERSION=2.28.3-1 RUN git clone https://github.com/NVIDIA/nccl.git --branch v$LIBNCCL_VERSION /tmp/nccl \ && cd /tmp/nccl \ && make -j $(nproc) \ @@ -119,7 +119,7 @@ RUN git clone https://github.com/NVIDIA/nccl.git --branch v$LIBNCCL_VERSION /tmp && cd && rm -rf /tmp/nccl # Install AWS-OFI-NCCL plugin -ARG AWS_OFI_NCCL_VERSION=1.16.3 +ARG AWS_OFI_NCCL_VERSION=1.17.0 RUN curl -sL https://github.com/aws/aws-ofi-nccl/releases/download/v$AWS_OFI_NCCL_VERSION/aws-ofi-nccl-$AWS_OFI_NCCL_VERSION.tar.gz | tar xvz -C /tmp \ && cd /tmp/aws-ofi-nccl-$AWS_OFI_NCCL_VERSION \ && ./configure \ diff --git a/test/images/nvidia/Dockerfile b/test/images/nvidia/Dockerfile index 1e5bfed48..02da98736 100644 --- a/test/images/nvidia/Dockerfile +++ b/test/images/nvidia/Dockerfile @@ -72,7 +72,7 @@ RUN git clone https://github.com/NVIDIA/nvbandwidth.git --branch $NVBANDWIDTH_VE && cd && rm -rf /tmp/cuda-samples # Install NCCL -ARG LIBNCCL_VERSION=2.27.7-1 +ARG LIBNCCL_VERSION=2.28.3-1 RUN git clone https://github.com/NVIDIA/nccl.git --branch v$LIBNCCL_VERSION /tmp/nccl \ && cd /tmp/nccl \ && make -j $(nproc) \ @@ -80,7 +80,7 @@ RUN git clone https://github.com/NVIDIA/nccl.git --branch v$LIBNCCL_VERSION /tmp && cd && rm -rf /tmp/nccl # Install AWS-OFI-NCCL plugin -ARG AWS_OFI_NCCL_VERSION=1.16.3 +ARG AWS_OFI_NCCL_VERSION=1.17.0 RUN curl -sL https://github.com/aws/aws-ofi-nccl/releases/download/v$AWS_OFI_NCCL_VERSION/aws-ofi-nccl-$AWS_OFI_NCCL_VERSION.tar.gz | tar xvz -C /tmp \ && cd /tmp/aws-ofi-nccl-$AWS_OFI_NCCL_VERSION \ && ./configure \