Skip to content

Commit 4973375

Browse files
github-actions[bot]web-flow
authored andcommitted
chore: update nvidia test dependencies (aws#658)
Co-authored-by: GitHub <[email protected]>
1 parent dd043df commit 4973375

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

test/images/nvidia-training/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,15 +111,15 @@ RUN curl -sL https://efa-installer.amazonaws.com/aws-efa-installer-$EFA_INSTALLE
111111
&& cd && rm -rf /tmp/aws-efa-installer
112112

113113
# Install NCCL
114-
ARG LIBNCCL_VERSION=2.27.5-1
114+
ARG LIBNCCL_VERSION=2.27.6-1
115115
RUN git clone https://github.com/NVIDIA/nccl.git --branch v$LIBNCCL_VERSION /tmp/nccl \
116116
&& cd /tmp/nccl \
117117
&& make -j $(nproc) \
118118
&& make install \
119119
&& cd && rm -rf /tmp/nccl
120120

121121
# Install AWS-OFI-NCCL plugin
122-
ARG AWS_OFI_NCCL_VERSION=1.16.0
122+
ARG AWS_OFI_NCCL_VERSION=1.16.2
123123
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 \
124124
&& cd /tmp/aws-ofi-nccl-$AWS_OFI_NCCL_VERSION \
125125
&& ./configure \

test/images/nvidia/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,15 +72,15 @@ RUN git clone https://github.com/NVIDIA/nvbandwidth.git --branch $NVBANDWIDTH_VE
7272
&& cd && rm -rf /tmp/cuda-samples
7373

7474
# Install NCCL
75-
ARG LIBNCCL_VERSION=2.27.5-1
75+
ARG LIBNCCL_VERSION=2.27.6-1
7676
RUN git clone https://github.com/NVIDIA/nccl.git --branch v$LIBNCCL_VERSION /tmp/nccl \
7777
&& cd /tmp/nccl \
7878
&& make -j $(nproc) \
7979
&& make install \
8080
&& cd && rm -rf /tmp/nccl
8181

8282
# Install AWS-OFI-NCCL plugin
83-
ARG AWS_OFI_NCCL_VERSION=1.16.0
83+
ARG AWS_OFI_NCCL_VERSION=1.16.2
8484
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 \
8585
&& cd /tmp/aws-ofi-nccl-$AWS_OFI_NCCL_VERSION \
8686
&& ./configure \

0 commit comments

Comments
 (0)