Skip to content

Commit b2d9318

Browse files
CamyllCamyll
committed
change python and cuda version
1 parent 0c5ce4c commit b2d9318

8 files changed

Lines changed: 399 additions & 357 deletions

‎tools/scripts/generate_binary_build_matrix.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -22,14 +22,14 @@
2222

2323

2424
PYTHON_ARCHES_DICT = {
25-
"nightly": ["3.10", "3.11", "3.12", "3.13", "3.13t", "3.14", "3.14t"],
26-
"test": ["3.10", "3.11", "3.12", "3.13", "3.13t", "3.14", "3.14t"],
27-
"release": ["3.9", "3.10", "3.11", "3.12", "3.13", "3.13t"],
25+
"nightly": ["3.10", "3.11", "3.12", "3.13", "3.13t", "3.14"],
26+
"test": ["3.10", "3.11", "3.12", "3.13", "3.13t", "3.14"],
27+
"release": ["3.10", "3.11", "3.12", "3.13", "3.14"],
2828
}
2929
CUDA_ARCHES_DICT = {
30-
"nightly": ["12.6", "12.8", "13.0"],
31-
"test": ["12.6", "12.8", "13.0"],
32-
"release": ["12.6", "12.8", "12.9"],
30+
"nightly": ["12.6", "12.8", "12.9", "13.0"],
31+
"test": ["12.6", "12.8", "12.9", "13.0"],
32+
"release": ["12.6", "12.8", "12.9", "13.0"],
3333
}
3434

3535
ROCM_ARCHES_DICT = {

‎tools/tests/assets/build_matrix_linux_wheel_cuda.json‎

Lines changed: 80 additions & 80 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,20 @@
4242
"upload_to_base_bucket": "no",
4343
"stable_version": "2.9.0"
4444
},
45+
{
46+
"python_version": "3.10",
47+
"gpu_arch_type": "cuda",
48+
"gpu_arch_version": "12.9",
49+
"desired_cuda": "cu129",
50+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
51+
"package_type": "manywheel",
52+
"build_name": "manywheel-py3_10-cuda12_9",
53+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
54+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
55+
"channel": "nightly",
56+
"upload_to_base_bucket": "no",
57+
"stable_version": "2.9.0"
58+
},
4559
{
4660
"python_version": "3.10",
4761
"gpu_arch_type": "cuda",
@@ -126,6 +140,20 @@
126140
"upload_to_base_bucket": "no",
127141
"stable_version": "2.9.0"
128142
},
143+
{
144+
"python_version": "3.11",
145+
"gpu_arch_type": "cuda",
146+
"gpu_arch_version": "12.9",
147+
"desired_cuda": "cu129",
148+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
149+
"package_type": "manywheel",
150+
"build_name": "manywheel-py3_11-cuda12_9",
151+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
152+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
153+
"channel": "nightly",
154+
"upload_to_base_bucket": "no",
155+
"stable_version": "2.9.0"
156+
},
129157
{
130158
"python_version": "3.11",
131159
"gpu_arch_type": "cuda",
@@ -210,6 +238,20 @@
210238
"upload_to_base_bucket": "no",
211239
"stable_version": "2.9.0"
212240
},
241+
{
242+
"python_version": "3.12",
243+
"gpu_arch_type": "cuda",
244+
"gpu_arch_version": "12.9",
245+
"desired_cuda": "cu129",
246+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
247+
"package_type": "manywheel",
248+
"build_name": "manywheel-py3_12-cuda12_9",
249+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
250+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
251+
"channel": "nightly",
252+
"upload_to_base_bucket": "no",
253+
"stable_version": "2.9.0"
254+
},
213255
{
214256
"python_version": "3.12",
215257
"gpu_arch_type": "cuda",
@@ -294,6 +336,20 @@
294336
"upload_to_base_bucket": "no",
295337
"stable_version": "2.9.0"
296338
},
339+
{
340+
"python_version": "3.13",
341+
"gpu_arch_type": "cuda",
342+
"gpu_arch_version": "12.9",
343+
"desired_cuda": "cu129",
344+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
345+
"package_type": "manywheel",
346+
"build_name": "manywheel-py3_13-cuda12_9",
347+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
348+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
349+
"channel": "nightly",
350+
"upload_to_base_bucket": "no",
351+
"stable_version": "2.9.0"
352+
},
297353
{
298354
"python_version": "3.13",
299355
"gpu_arch_type": "cuda",
@@ -378,6 +434,20 @@
378434
"upload_to_base_bucket": "no",
379435
"stable_version": "2.9.0"
380436
},
437+
{
438+
"python_version": "3.13t",
439+
"gpu_arch_type": "cuda",
440+
"gpu_arch_version": "12.9",
441+
"desired_cuda": "cu129",
442+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
443+
"package_type": "manywheel",
444+
"build_name": "manywheel-py3_13t-cuda12_9",
445+
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
446+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
447+
"channel": "nightly",
448+
"upload_to_base_bucket": "no",
449+
"stable_version": "2.9.0"
450+
},
381451
{
382452
"python_version": "3.13t",
383453
"gpu_arch_type": "cuda",
@@ -465,123 +535,53 @@
465535
{
466536
"python_version": "3.14",
467537
"gpu_arch_type": "cuda",
468-
"gpu_arch_version": "13.0",
469-
"desired_cuda": "cu130",
470-
"container_image": "pytorch/manylinux2_28-builder:cuda13.0",
538+
"gpu_arch_version": "12.9",
539+
"desired_cuda": "cu129",
540+
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
471541
"package_type": "manywheel",
472-
"build_name": "manywheel-py3_14-cuda13_0",
542+
"build_name": "manywheel-py3_14-cuda12_9",
473543
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
474-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130",
475-
"channel": "nightly",
476-
"upload_to_base_bucket": "no",
477-
"stable_version": "2.9.0"
478-
},
479-
{
480-
"python_version": "3.14",
481-
"gpu_arch_type": "rocm",
482-
"gpu_arch_version": "6.4",
483-
"desired_cuda": "rocm6.4",
484-
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
485-
"package_type": "manywheel",
486-
"build_name": "manywheel-py3_14-rocm6_4",
487-
"validation_runner": "linux.2xlarge",
488-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
544+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
489545
"channel": "nightly",
490546
"upload_to_base_bucket": "no",
491547
"stable_version": "2.9.0"
492548
},
493549
{
494550
"python_version": "3.14",
495-
"gpu_arch_type": "rocm",
496-
"gpu_arch_version": "7.0",
497-
"desired_cuda": "rocm7.0",
498-
"container_image": "pytorch/manylinux2_28-builder:rocm7.0",
499-
"package_type": "manywheel",
500-
"build_name": "manywheel-py3_14-rocm7_0",
501-
"validation_runner": "linux.2xlarge",
502-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm7.0",
503-
"channel": "nightly",
504-
"upload_to_base_bucket": "no",
505-
"stable_version": "2.9.0"
506-
},
507-
{
508-
"python_version": "3.14t",
509-
"gpu_arch_type": "cpu",
510-
"gpu_arch_version": "",
511-
"desired_cuda": "cpu",
512-
"container_image": "pytorch/manylinux2_28-builder:cpu",
513-
"package_type": "manywheel",
514-
"build_name": "manywheel-py3_14t-cpu",
515-
"validation_runner": "linux.2xlarge",
516-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu",
517-
"channel": "nightly",
518-
"upload_to_base_bucket": "no",
519-
"stable_version": "2.9.0"
520-
},
521-
{
522-
"python_version": "3.14t",
523-
"gpu_arch_type": "cuda",
524-
"gpu_arch_version": "12.6",
525-
"desired_cuda": "cu126",
526-
"container_image": "pytorch/manylinux2_28-builder:cuda12.6",
527-
"package_type": "manywheel",
528-
"build_name": "manywheel-py3_14t-cuda12_6",
529-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
530-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126",
531-
"channel": "nightly",
532-
"upload_to_base_bucket": "no",
533-
"stable_version": "2.9.0"
534-
},
535-
{
536-
"python_version": "3.14t",
537-
"gpu_arch_type": "cuda",
538-
"gpu_arch_version": "12.8",
539-
"desired_cuda": "cu128",
540-
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
541-
"package_type": "manywheel",
542-
"build_name": "manywheel-py3_14t-cuda12_8",
543-
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
544-
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128",
545-
"channel": "nightly",
546-
"upload_to_base_bucket": "no",
547-
"stable_version": "2.9.0"
548-
},
549-
{
550-
"python_version": "3.14t",
551551
"gpu_arch_type": "cuda",
552552
"gpu_arch_version": "13.0",
553553
"desired_cuda": "cu130",
554554
"container_image": "pytorch/manylinux2_28-builder:cuda13.0",
555555
"package_type": "manywheel",
556-
"build_name": "manywheel-py3_14t-cuda13_0",
556+
"build_name": "manywheel-py3_14-cuda13_0",
557557
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
558558
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130",
559559
"channel": "nightly",
560560
"upload_to_base_bucket": "no",
561561
"stable_version": "2.9.0"
562562
},
563563
{
564-
"python_version": "3.14t",
564+
"python_version": "3.14",
565565
"gpu_arch_type": "rocm",
566566
"gpu_arch_version": "6.4",
567567
"desired_cuda": "rocm6.4",
568568
"container_image": "pytorch/manylinux2_28-builder:rocm6.4",
569569
"package_type": "manywheel",
570-
"build_name": "manywheel-py3_14t-rocm6_4",
570+
"build_name": "manywheel-py3_14-rocm6_4",
571571
"validation_runner": "linux.2xlarge",
572572
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4",
573573
"channel": "nightly",
574574
"upload_to_base_bucket": "no",
575575
"stable_version": "2.9.0"
576576
},
577577
{
578-
"python_version": "3.14t",
578+
"python_version": "3.14",
579579
"gpu_arch_type": "rocm",
580580
"gpu_arch_version": "7.0",
581581
"desired_cuda": "rocm7.0",
582582
"container_image": "pytorch/manylinux2_28-builder:rocm7.0",
583583
"package_type": "manywheel",
584-
"build_name": "manywheel-py3_14t-rocm7_0",
584+
"build_name": "manywheel-py3_14-rocm7_0",
585585
"validation_runner": "linux.2xlarge",
586586
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm7.0",
587587
"channel": "nightly",

0 commit comments

Comments
 (0)