We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3c893d2 commit 8988e88Copy full SHA for 8988e88
.github/scripts/install-cuda-aarch64.sh
@@ -12,7 +12,9 @@ install_cuda_aarch64() {
12
# nccl version must match libtorch_cuda.so was built with
13
if [[ ${CU_VERSION:0:4} == "cu12" ]]; then
14
# cu12: https://github.com/pytorch/pytorch/blob/main/.ci/docker/ci_commit_pins/nccl-cu12.txt
15
- if [[ ${CU_VERSION} == "cu128" ]]; then
+ if [[ ${CU_VERSION} == "cu129" ]]; then
16
+ nccl_version="2.27.5-1"
17
+ elif [[ ${CU_VERSION} == "cu128" ]]; then
18
nccl_version="2.26.2-1"
19
elif [[ ${CU_VERSION} == "cu126" ]]; then
20
nccl_version="2.24.3-1"
0 commit comments