Skip to content

Commit

Permalink
Revise nccl versions and switch to object from s3.
Browse files Browse the repository at this point in the history
  • Loading branch information
wbrown committed Nov 29, 2023
1 parent 2f6c789 commit 03f9015
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions .github/workflows/ubuntu-20.yml
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ jobs:
base-tag: 11.8.0-cudnn8-devel-ubuntu20.04
cuda-version-minor: "11.8.0"
cuda-version-major: "11.8"
nccl-version: 2.16.2-1
nccl-version: 2.16.5-1
cuda-samples-version: "11.6"
hpcx-version: "2.14"
hpcx-nccl-version: "2.16"
Expand All @@ -32,7 +32,7 @@ jobs:
base-tag: 12.0.1-cudnn8-devel-ubuntu20.04
cuda-version-minor: "12.0.1"
cuda-version-major: "12.0"
nccl-version: 2.18.5-1
nccl-version: 2.19.3-1
cuda-samples-version: "12.0"
hpcx-version: "2.16"
hpcx-nccl-version: "2.18"
Expand Down Expand Up @@ -64,7 +64,7 @@ jobs:
base-tag: 12.2.2-cudnn8-devel-ubuntu20.04
cuda-version-minor: "12.2.2"
cuda-version-major: "12.2"
nccl-version: 2.18.5-1
nccl-version: 2.19.3-1
cuda-samples-version: "12.2"
hpcx-version: "2.16"
hpcx-nccl-version: "2.18"
Expand Down
2 changes: 1 addition & 1 deletion Dockerfile.ubuntu20
Original file line number Diff line number Diff line change
Expand Up @@ -67,7 +67,7 @@ ARG HPCX_MLNX_OFED="MLNX_OFED_LINUX-5"
RUN cd /tmp && \
export HPCX_DISTRIBUTION="hpcx-v${HPCX_VERSION}-gcc-${HPCX_MLNX_OFED}-ubuntu20.04-cuda${HPCX_CUDA_VERSION}-gdrcopy2-nccl${HPCX_NCCL_VERSION}-x86_64" \
HPCX_DIR="/opt/hpcx" && \
wget -q -O - http://blobstore.s3.ord1.coreweave.com/drivers/${HPCX_DISTRIBUTION}.tbz | tar xjf - && \
wget -q -O - http://blobstore.object.ord1.coreweave.com/drivers/${HPCX_DISTRIBUTION}.tbz | tar xjf - && \
grep -IrlF "/build-result/${HPCX_DISTRIBUTION}" ${HPCX_DISTRIBUTION} | xargs -rd'\n' sed -i -e "s:/build-result/${HPCX_DISTRIBUTION}:${HPCX_DIR}:g" && \
mv ${HPCX_DISTRIBUTION} ${HPCX_DIR}

Expand Down

0 comments on commit 03f9015

Please sign in to comment.