Skip to content

Commit

Permalink
Change nccl version to 2.20.3
Browse files Browse the repository at this point in the history
NCCL 2.19.4 has performance regression.
  • Loading branch information
mhuguesaws authored Mar 21, 2024
1 parent 20bc096 commit 7f9aa44
Showing 1 changed file with 3 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -2,10 +2,10 @@
# SPDX-License-Identifier: MIT-0
FROM nvidia/cuda:12.2.0-devel-ubuntu22.04

ARG EFA_INSTALLER_VERSION=1.30.0
ARG EFA_INSTALLER_VERSION=1.31.0
ARG AWS_OFI_NCCL_VERSION=v1.8.1-aws
ARG NCCL_TESTS_VERSION=2.13.9
ARG NCCL_VERSION=2.19.4
ARG NCCL_VERSION=2.20.3

RUN apt-get update -y
RUN apt-get remove -y --allow-change-held-packages \
Expand Down Expand Up @@ -102,4 +102,4 @@ RUN git clone -b v${NCCL_TESTS_VERSION} https://github.com/NVIDIA/nccl-tests.git
NVCC_GENCODE="-gencode=arch=compute_80,code=sm_80 -gencode=arch=compute_86,code=sm_86 -gencode=arch=compute_90,code=sm_90"

RUN rm -rf /var/lib/apt/lists/*
ENV LD_PRELOAD /opt/nccl/build/lib/libnccl.so
ENV LD_PRELOAD /opt/nccl/build/lib/libnccl.so

0 comments on commit 7f9aa44

Please sign in to comment.