Skip to content

Commit ba525d4

Browse files
authored
chore: update nvidia test dependencies
1 parent 0da3c47 commit ba525d4

2 files changed

Lines changed: 5 additions & 5 deletions

File tree

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.28.7-1
114+
ARG LIBNCCL_VERSION=2.27.3-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.17.2
122+
ARG AWS_OFI_NCCL_VERSION=1.19.1
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: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -66,23 +66,23 @@ RUN curl -sL https://efa-installer.amazonaws.com/aws-efa-installer-$EFA_INSTALLE
6666
&& cd && rm -rf /tmp/aws-efa-installer
6767

6868
# Build nvbandwidth
69-
ARG NVBANDWIDTH_VERSION=v0.8
69+
ARG NVBANDWIDTH_VERSION=v0.9
7070
RUN apt install -y libboost-program-options-dev
7171
RUN git clone https://github.com/NVIDIA/nvbandwidth.git --branch $NVBANDWIDTH_VERSION /tmp/nvbandwidth \
7272
&& cd /tmp/nvbandwidth \
7373
&& cmake -DMULTINODE=1 . && make && cp nvbandwidth /usr/bin \
7474
&& cd && rm -rf /tmp/cuda-samples
7575

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

8484
# Install AWS-OFI-NCCL plugin
85-
ARG AWS_OFI_NCCL_VERSION=1.17.2
85+
ARG AWS_OFI_NCCL_VERSION=1.19.1
8686
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 \
8787
&& cd /tmp/aws-ofi-nccl-$AWS_OFI_NCCL_VERSION \
8888
&& ./configure \

0 commit comments

Comments
 (0)