8000 Remove Cuda 12.4 from nightly Binaries (#148625) · pytorch/pytorch@1239176 · GitHub
[go: up one dir, main page]
Skip to content

Commit 1239176

Browse files
tinglvvpytorchmergebot
tinglvv
authored and
pytorchmergebot
committed
Remove Cuda 12.4 from nightly Binaries (#148625)
#145570 removes cuda 12.4 nightly builds Pull Request resolved: #148625 Approved by: https://github.com/atalman
1 parent 61c4074 commit 1239176

11 files changed

+484
-2954
lines changed

‎.github/scripts/generate_binary_build_matrix.py

Lines changed: 6 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -16,16 +16,14 @@
1616

1717

1818
# NOTE: Also update the CUDA sources in tools/nightly.py when changing this list
19-
CUDA_ARCHES = ["11.8", "12.4", "12.6", "12.8"]
19+
CUDA_ARCHES = ["11.8", "12.6", "12.8"]
2020
CUDA_ARCHES_FULL_VERSION = {
2121
"11.8": "11.8.0",
22-
"12.4": "12.4.1",
2322
"12.6": "12.6.3",
2423
"12.8": "12.8.0",
2524
}
2625
CUDA_ARCHES_CUDNN_VERSION = {
2726
"11.8": "9",
28-
"12.4": "9",
2927
"12.6": "9",
3028
"12.8": "9",
3129
}
@@ -58,21 +56,6 @@
5856
"nvidia-nccl-cu11==2.21.5; platform_system == 'Linux' and platform_machine == 'x86_64' | "
5957
"nvidia-nvtx-cu11==11.8.86; platform_system == 'Linux' and platform_machine == 'x86_64'"
6058
),
61-
"12.4": (
62-
"nvidia-cuda-nvrtc-cu12==12.4.127; platform_system == 'Linux' and platform_machine == 'x86_64' | "
63-
"nvidia-cuda-runtime-cu12==12.4.127; platform_system == 'Linux' and platform_machine == 'x86_64' | "
64-
"nvidia-cuda-cupti-cu12==12.4.127; platform_system == 'Linux' and platform_machine == 'x86_64' | "
65-
"nvidia-cudnn-cu12==9.1.0.70; platform_system == 'Linux' and platform_machine == 'x86_64' | "
66-
"nvidia-cublas-cu12==12.4.5.8; platform_system == 'Linux' and platform_machine == 'x86_64' | "
67-
"nvidia-cufft-cu12==11.2.1.3; platform_system == 'Linux' and platform_machine == 'x86_64' | "
68-
"nvidia-curand-cu12==10.3.5.147; platform_system == 'Linux' and platform_machine == 'x86_64' | "
69-
"nvidia-cusolver-cu12==11.6.1.9; platform_system == 'Linux' and platform_machine == 'x86_64' | "
70-
"nvidia-cusparse-cu12==12.3.1.170; platform_system == 'Linux' and platform_machine == 'x86_64' | "
71-
"nvidia-cusparselt-cu12==0.6.2; platform_system == 'Linux' and platform_machine == 'x86_64' | "
72-
"nvidia-nccl-cu12==2.25.1; platform_system == 'Linux' and platform_machine == 'x86_64' | "
73-
"nvidia-nvtx-cu12==12.4.127; platform_system == 'Linux' and platform_machine == 'x86_64' | "
74-
"nvidia-nvjitlink-cu12==12.4.127; platform_system == 'Linux' and platform_machine == 'x86_64'"
75-
),
7659
"12.6": (
7760
"nvidia-cuda-nvrtc-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7861
"nvidia-cuda-runtime-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "
@@ -348,7 +331,7 @@ def generate_wheels_matrix(
348331
continue
349332

350333
if use_split_build and (
351-
arch_version not in ["12.6", "12.4", "11.8", "cpu"] or os != "linux"
334+
arch_version not in ["12.6", "12.8", "11.8", "cpu"] or os != "linux"
352335
):
353336
raise RuntimeError(
354337
"Split build is only supported on linux with cuda 12*, 11.8, and cpu.\n"
@@ -359,7 +342,7 @@ def generate_wheels_matrix(
359342
# cuda linux wheels require PYTORCH_EXTRA_INSTALL_REQUIREMENTS to install
360343

361344
if (
362-
arch_version in ["12.8", "12.6", "12.4", "11.8"]
345+
arch_version in ["12.8", "12.6", "11.8"]
363346
and os == "linux"
364347
or arch_version in CUDA_AARCH64_ARCHES
365348
):
@@ -388,8 +371,8 @@ def generate_wheels_matrix(
388371
), # include special case for aarch64 build, remove the -aarch64 postfix
389372
}
390373
)
391-
# Special build building to use on Colab. Python 3.11 for 12.4 CUDA
392-
if python_version == "3.11" and arch_version == "12.4":
374+
# Special build building to use on Colab. Python 3.11 for 12.6 CUDA
375+
if python_version == "3.11" and arch_version == "12.6":
393376
ret.append(
394377
{
395378
"python_version": python_version,
@@ -432,7 +415,7 @@ def generate_wheels_matrix(
432415
"pytorch_extra_install_requirements": (
433416
PYTORCH_EXTRA_INSTALL_REQUIREMENTS["xpu"]
434417
if gpu_arch_type == "xpu"
435-
else PYTORCH_EXTRA_INSTALL_REQUIREMENTS["12.4"]
418+
else PYTORCH_EXTRA_INSTALL_REQUIREMENTS["12.6"]
436419
if os != "linux"
437420
else ""
438421
),
@@ -444,5 +427,4 @@ def generate_wheels_matrix(
444427

445428
validate_nccl_dep_consistency("12.8")
446429
validate_nccl_dep_consistency("12.6")
447-
validate_nccl_dep_consistency("12.4")
448430
validate_nccl_dep_consistency("11.8")

‎.github/scripts/generate_ci_workflows.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ class OperatingSystem:
152152
package_type="manywheel",
153153
build_configs=generate_binary_build_matrix.generate_wheels_matrix(
154154
OperatingSystem.LINUX,
155-
arches=["11.8", "12.4", "12.6", "12.8"],
155+
arches=["11.8", "12.6", "12.8"],
156156
python_versions=["3.9"],
157157
),
158158
branches="main",

0 commit comments

Comments
 (0)
0