|
1 | 1 | { |
2 | 2 | "include": [ |
3 | | - { |
4 | | - "python_version": "3.9", |
5 | | - "gpu_arch_type": "cpu", |
6 | | - "gpu_arch_version": "", |
7 | | - "desired_cuda": "cpu", |
8 | | - "container_image": "pytorch/manylinux2_28-builder:cpu", |
9 | | - "package_type": "manywheel", |
10 | | - "build_name": "manywheel-py3_9-cpu", |
11 | | - "validation_runner": "linux.2xlarge", |
12 | | - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu", |
13 | | - "channel": "nightly", |
14 | | - "upload_to_base_bucket": "no", |
15 | | - "stable_version": "2.8.0", |
16 | | - "use_split_build": false |
17 | | - }, |
18 | | - { |
19 | | - "python_version": "3.9", |
20 | | - "gpu_arch_type": "cuda", |
21 | | - "gpu_arch_version": "12.6", |
22 | | - "desired_cuda": "cu126", |
23 | | - "container_image": "pytorch/manylinux2_28-builder:cuda12.6", |
24 | | - "package_type": "manywheel", |
25 | | - "build_name": "manywheel-py3_9-cuda12_6", |
26 | | - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", |
27 | | - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu126", |
28 | | - "channel": "nightly", |
29 | | - "upload_to_base_bucket": "no", |
30 | | - "stable_version": "2.8.0", |
31 | | - "use_split_build": false |
32 | | - }, |
33 | | - { |
34 | | - "python_version": "3.9", |
35 | | - "gpu_arch_type": "cuda", |
36 | | - "gpu_arch_version": "12.8", |
37 | | - "desired_cuda": "cu128", |
38 | | - "container_image": "pytorch/manylinux2_28-builder:cuda12.8", |
39 | | - "package_type": "manywheel", |
40 | | - "build_name": "manywheel-py3_9-cuda12_8", |
41 | | - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", |
42 | | - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128", |
43 | | - "channel": "nightly", |
44 | | - "upload_to_base_bucket": "no", |
45 | | - "stable_version": "2.8.0", |
46 | | - "use_split_build": false |
47 | | - }, |
48 | | - { |
49 | | - "python_version": "3.9", |
50 | | - "gpu_arch_type": "cuda", |
51 | | - "gpu_arch_version": "12.9", |
52 | | - "desired_cuda": "cu129", |
53 | | - "container_image": "pytorch/manylinux2_28-builder:cuda12.9", |
54 | | - "package_type": "manywheel", |
55 | | - "build_name": "manywheel-py3_9-cuda12_9", |
56 | | - "validation_runner": "linux.g5.4xlarge.nvidia.gpu", |
57 | | - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129", |
58 | | - "channel": "nightly", |
59 | | - "upload_to_base_bucket": "no", |
60 | | - "stable_version": "2.8.0", |
61 | | - "use_split_build": false |
62 | | - }, |
63 | | - { |
64 | | - "python_version": "3.9", |
65 | | - "gpu_arch_type": "rocm", |
66 | | - "gpu_arch_version": "6.3", |
67 | | - "desired_cuda": "rocm6.3", |
68 | | - "container_image": "pytorch/manylinux2_28-builder:rocm6.3", |
69 | | - "package_type": "manywheel", |
70 | | - "build_name": "manywheel-py3_9-rocm6_3", |
71 | | - "validation_runner": "linux.2xlarge", |
72 | | - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.3", |
73 | | - "channel": "nightly", |
74 | | - "upload_to_base_bucket": "no", |
75 | | - "stable_version": "2.8.0", |
76 | | - "use_split_build": false |
77 | | - }, |
78 | | - { |
79 | | - "python_version": "3.9", |
80 | | - "gpu_arch_type": "rocm", |
81 | | - "gpu_arch_version": "6.4", |
82 | | - "desired_cuda": "rocm6.4", |
83 | | - "container_image": "pytorch/manylinux2_28-builder:rocm6.4", |
84 | | - "package_type": "manywheel", |
85 | | - "build_name": "manywheel-py3_9-rocm6_4", |
86 | | - "validation_runner": "linux.2xlarge", |
87 | | - "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.4", |
88 | | - "channel": "nightly", |
89 | | - "upload_to_base_bucket": "no", |
90 | | - "stable_version": "2.8.0", |
91 | | - "use_split_build": false |
92 | | - }, |
93 | 3 | { |
94 | 4 | "python_version": "3.10", |
95 | 5 | "gpu_arch_type": "cpu", |
|
0 commit comments