Skip to content

Commit 8988e88

Browse files
add nccl 129 support (#3882)
1 parent 3c893d2 commit 8988e88

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

.github/scripts/install-cuda-aarch64.sh

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,9 @@ install_cuda_aarch64() {
1212
# nccl version must match libtorch_cuda.so was built with
1313
if [[ ${CU_VERSION:0:4} == "cu12" ]]; then
1414
# cu12: https://github.com/pytorch/pytorch/blob/main/.ci/docker/ci_commit_pins/nccl-cu12.txt
15-
if [[ ${CU_VERSION} == "cu128" ]]; then
15+
if [[ ${CU_VERSION} == "cu129" ]]; then
16+
nccl_version="2.27.5-1"
17+
elif [[ ${CU_VERSION} == "cu128" ]]; then
1618
nccl_version="2.26.2-1"
1719
elif [[ ${CU_VERSION} == "cu126" ]]; then
1820
nccl_version="2.24.3-1"

0 commit comments

Comments
 (0)