Skip to content

Commit a259b2e

Browse files
atalmanatalman
authored
Use Manylinux 2.28 for wheel build (#6086)
Switch nightly ecosystem Wheel Builds to manylinux 2.28 pytorch/pytorch PR: pytorch/pytorch#143423
1 parent 27d47f2 commit a259b2e

8 files changed

+80
-80
lines changed

‎tools/scripts/generate_binary_build_matrix.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -151,15 +151,15 @@ def initialize_globals(channel: str, build_python_only: bool) -> None:
151151
else:
152152
PYTHON_ARCHES = PYTHON_ARCHES_DICT[channel]
153153
WHEEL_CONTAINER_IMAGES = {
154-
"11.8": "pytorch/manylinux-builder:cuda11.8",
155-
"12.1": "pytorch/manylinux-builder:cuda12.1",
156-
"12.4": "pytorch/manylinux-builder:cuda12.4",
157-
"12.6": "pytorch/manylinux2_28-builder:cuda12.6",
154+
**{
155+
gpu_arch: f"pytorch/manylinux2_28-builder:cuda{gpu_arch}"
156+
for gpu_arch in CUDA_ARCHES
157+
},
158158
**{
159159
gpu_arch: f"pytorch/manylinux2_28-builder:rocm{gpu_arch}"
160160
for gpu_arch in ROCM_ARCHES
161161
},
162-
CPU: "pytorch/manylinux-builder:cpu",
162+
CPU: "pytorch/manylinux2_28-builder:cpu",
163163
XPU: "pytorch/manylinux2_28-builder:xpu",
164164
# TODO: Migrate CUDA_AARCH64 image to manylinux2_28_aarch64-builder:cuda12.4
165165
CPU_AARCH64: "pytorch/manylinux2_28_aarch64-builder:cpu-aarch64",

‎tools/tests/assets/build_matrix_linux_wheel_cuda.json

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
"gpu_arch_type": "cpu",
66
"gpu_arch_version": "",
77
"desired_cuda": "cpu",
8-
"container_image": "pytorch/manylinux-builder:cpu",
8+
"container_image": "pytorch/manylinux2_28-builder:cpu",
99
"package_type": "manywheel",
1010
"build_name": "manywheel-py3_9-cpu",
1111
"validation_runner": "linux.2xlarge",
@@ -20,7 +20,7 @@
2020
"gpu_arch_type": "cuda",
2121
"gpu_arch_version": "11.8",
2222
"desired_cuda": "cu118",
23-
"container_image": "pytorch/manylinux-builder:cuda11.8",
23+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
2424
"package_type": "manywheel",
2525
"build_name": "manywheel-py3_9-cuda11_8",
2626
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -35,7 +35,7 @@
3535
"gpu_arch_type": "cuda",
3636
"gpu_arch_version": "12.4",
3737
"desired_cuda": "cu124",
38-
"container_image": "pytorch/manylinux-builder:cuda12.4",
38+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
3939
"package_type": "manywheel",
4040
"build_name": "manywheel-py3_9-cuda12_4",
4141
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -95,7 +95,7 @@
9595
"gpu_arch_type": "cpu",
9696
"gpu_arch_version": "",
9797
"desired_cuda": "cpu",
98-
"container_image": "pytorch/manylinux-builder:cpu",
98+
"container_image": "pytorch/manylinux2_28-builder:cpu",
9999
"package_type": "manywheel",
100100
"build_name": "manywheel-py3_10-cpu",
101101
"validation_runner": "linux.2xlarge",
@@ -110,7 +110,7 @@
110110
"gpu_arch_type": "cuda",
111111
"gpu_arch_version": "11.8",
112112
"desired_cuda": "cu118",
113-
"container_image": "pytorch/manylinux-builder:cuda11.8",
113+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
114114
"package_type": "manywheel",
115115
"build_name": "manywheel-py3_10-cuda11_8",
116116
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -125,7 +125,7 @@
125125
"gpu_arch_type": "cuda",
126126
"gpu_arch_version": "12.4",
127127
"desired_cuda": "cu124",
128-
"container_image": "pytorch/manylinux-builder:cuda12.4",
128+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
129129
"package_type": "manywheel",
130130
"build_name": "manywheel-py3_10-cuda12_4",
131131
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -185,7 +185,7 @@
185185
"gpu_arch_type": "cpu",
186186
"gpu_arch_version": "",
187187
"desired_cuda": "cpu",
188-
"container_image": "pytorch/manylinux-builder:cpu",
188+
"container_image": "pytorch/manylinux2_28-builder:cpu",
189189
"package_type": "manywheel",
190190
"build_name": "manywheel-py3_11-cpu",
191191
"validation_runner": "linux.2xlarge",
@@ -200,7 +200,7 @@
200200
"gpu_arch_type": "cuda",
201201
"gpu_arch_version": "11.8",
202202
"desired_cuda": "cu118",
203-
"container_image": "pytorch/manylinux-builder:cuda11.8",
203+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
204204
"package_type": "manywheel",
205205
"build_name": "manywheel-py3_11-cuda11_8",
206206
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -215,7 +215,7 @@
215215
"gpu_arch_type": "cuda",
216216
"gpu_arch_version": "12.4",
217217
"desired_cuda": "cu124",
218-
"container_image": "pytorch/manylinux-builder:cuda12.4",
218+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
219219
"package_type": "manywheel",
220220
"build_name": "manywheel-py3_11-cuda12_4",
221221
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -275,7 +275,7 @@
275275
"gpu_arch_type": "cpu",
276276
"gpu_arch_version": "",
277277
"desired_cuda": "cpu",
278-
"container_image": "pytorch/manylinux-builder:cpu",
278+
"container_image": "pytorch/manylinux2_28-builder:cpu",
279279
"package_type": "manywheel",
280280
"build_name": "manywheel-py3_12-cpu",
281281
"validation_runner": "linux.2xlarge",
@@ -290,7 +290,7 @@
290290
"gpu_arch_type": "cuda",
291291
"gpu_arch_version": "11.8",
292292
"desired_cuda": "cu118",
293-
"container_image": "pytorch/manylinux-builder:cuda11.8",
293+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
294294
"package_type": "manywheel",
295295
"build_name": "manywheel-py3_12-cuda11_8",
296296
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -305,7 +305,7 @@
305305
"gpu_arch_type": "cuda",
306306
"gpu_arch_version": "12.4",
307307
"desired_cuda": "cu124",
308-
"container_image": "pytorch/manylinux-builder:cuda12.4",
308+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
309309
"package_type": "manywheel",
310310
"build_name": "manywheel-py3_12-cuda12_4",
311311
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -365,7 +365,7 @@
365365
"gpu_arch_type": "cpu",
366366
"gpu_arch_version": "",
367367
"desired_cuda": "cpu",
368-
"container_image": "pytorch/manylinux-builder:cpu",
368+
"container_image": "pytorch/manylinux2_28-builder:cpu",
369369
"package_type": "manywheel",
370370
"build_name": "manywheel-py3_13-cpu",
371371
"validation_runner": "linux.2xlarge",
@@ -380,7 +380,7 @@
380380
"gpu_arch_type": "cuda",
381381
"gpu_arch_version": "11.8",
382382
"desired_cuda": "cu118",
383-
"container_image": "pytorch/manylinux-builder:cuda11.8",
383+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
384384
"package_type": "manywheel",
385385
"build_name": "manywheel-py3_13-cuda11_8",
386386
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -395,7 +395,7 @@
395395
"gpu_arch_type": "cuda",
396396
"gpu_arch_version": "12.4",
397397
"desired_cuda": "cu124",
398-
"container_image": "pytorch/manylinux-builder:cuda12.4",
398+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
399399
"package_type": "manywheel",
400400
"build_name": "manywheel-py3_13-cuda12_4",
401401
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",

‎tools/tests/assets/build_matrix_linux_wheel_cuda_norocm.json

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
"gpu_arch_type": "cpu",
66
"gpu_arch_version": "",
77
"desired_cuda": "cpu",
8-
"container_image": "pytorch/manylinux-builder:cpu",
8+
"container_image": "pytorch/manylinux2_28-builder:cpu",
99
"package_type": "manywheel",
1010
"build_name": "manywheel-py3_9-cpu",
1111
"validation_runner": "linux.2xlarge",
@@ -20,7 +20,7 @@
2020
"gpu_arch_type": "cuda",
2121
"gpu_arch_version": "11.8",
2222
"desired_cuda": "cu118",
23-
"container_image": "pytorch/manylinux-builder:cuda11.8",
23+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
2424
"package_type": "manywheel",
2525
"build_name": "manywheel-py3_9-cuda11_8",
2626
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -35,7 +35,7 @@
3535
"gpu_arch_type": "cuda",
3636
"gpu_arch_version": "12.4",
3737
"desired_cuda": "cu124",
38-
"container_image": "pytorch/manylinux-builder:cuda12.4",
38+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
3939
"package_type": "manywheel",
4040
"build_name": "manywheel-py3_9-cuda12_4",
4141
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -65,7 +65,7 @@
6565
"gpu_arch_type": "cpu",
6666
"gpu_arch_version": "",
6767
"desired_cuda": "cpu",
68-
"container_image": "pytorch/manylinux-builder:cpu",
68+
"container_image": "pytorch/manylinux2_28-builder:cpu",
6969
"package_type": "manywheel",
7070
"build_name": "manywheel-py3_10-cpu",
7171
"validation_runner": "linux.2xlarge",
@@ -80,7 +80,7 @@
8080
"gpu_arch_type": "cuda",
8181
"gpu_arch_version": "11.8",
8282
"desired_cuda": "cu118",
83-
"container_image": "pytorch/manylinux-builder:cuda11.8",
83+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
8484
"package_type": "manywheel",
8585
"build_name": "manywheel-py3_10-cuda11_8",
8686
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -95,7 +95,7 @@
9595
"gpu_arch_type": "cuda",
9696
"gpu_arch_version": "12.4",
9797
"desired_cuda": "cu124",
98-
"container_image": "pytorch/manylinux-builder:cuda12.4",
98+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
9999
"package_type": "manywheel",
100100
"build_name": "manywheel-py3_10-cuda12_4",
101101
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -125,7 +125,7 @@
125125
"gpu_arch_type": "cpu",
126126
"gpu_arch_version": "",
127127
"desired_cuda": "cpu",
128-
"container_image": "pytorch/manylinux-builder:cpu",
128+
"container_image": "pytorch/manylinux2_28-builder:cpu",
129129
"package_type": "manywheel",
130130
"build_name": "manywheel-py3_11-cpu",
131131
"validation_runner": "linux.2xlarge",
@@ -140,7 +140,7 @@
140140
"gpu_arch_type": "cuda",
141141
"gpu_arch_version": "11.8",
142142
"desired_cuda": "cu118",
143-
"container_image": "pytorch/manylinux-builder:cuda11.8",
143+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
144144
"package_type": "manywheel",
145145
"build_name": "manywheel-py3_11-cuda11_8",
146146
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -155,7 +155,7 @@
155155
"gpu_arch_type": "cuda",
156156
"gpu_arch_version": "12.4",
157157
"desired_cuda": "cu124",
158-
"container_image": "pytorch/manylinux-builder:cuda12.4",
158+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
159159
"package_type": "manywheel",
160160
"build_name": "manywheel-py3_11-cuda12_4",
161161
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -185,7 +185,7 @@
185185
"gpu_arch_type": "cpu",
186186
"gpu_arch_version": "",
187187
"desired_cuda": "cpu",
188-
"container_image": "pytorch/manylinux-builder:cpu",
188+
"container_image": "pytorch/manylinux2_28-builder:cpu",
189189
"package_type": "manywheel",
190190
"build_name": "manywheel-py3_12-cpu",
191191
"validation_runner": "linux.2xlarge",
@@ -200,7 +200,7 @@
200200
"gpu_arch_type": "cuda",
201201
"gpu_arch_version": "11.8",
202202
"desired_cuda": "cu118",
203-
"container_image": "pytorch/manylinux-builder:cuda11.8",
203+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
204204
"package_type": "manywheel",
205205
"build_name": "manywheel-py3_12-cuda11_8",
206206
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -215,7 +215,7 @@
215215
"gpu_arch_type": "cuda",
216216
"gpu_arch_version": "12.4",
217217
"desired_cuda": "cu124",
218-
"container_image": "pytorch/manylinux-builder:cuda12.4",
218+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
219219
"package_type": "manywheel",
220220
"build_name": "manywheel-py3_12-cuda12_4",
221221
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -245,7 +245,7 @@
245245
"gpu_arch_type": "cpu",
246246
"gpu_arch_version": "",
247247
"desired_cuda": "cpu",
248-
"container_image": "pytorch/manylinux-builder:cpu",
248+
"container_image": "pytorch/manylinux2_28-builder:cpu",
249249
"package_type": "manywheel",
250250
"build_name": "manywheel-py3_13-cpu",
251251
"validation_runner": "linux.2xlarge",
@@ -260,7 +260,7 @@
260260
"gpu_arch_type": "cuda",
261261
"gpu_arch_version": "11.8",
262262
"desired_cuda": "cu118",
263-
"container_image": "pytorch/manylinux-builder:cuda11.8",
263+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
264264
"package_type": "manywheel",
265265
"build_name": "manywheel-py3_13-cuda11_8",
266266
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -275,7 +275,7 @@
275275
"gpu_arch_type": "cuda",
276276
"gpu_arch_version": "12.4",
277277
"desired_cuda": "cu124",
278-
"container_image": "pytorch/manylinux-builder:cuda12.4",
278+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
279279
"package_type": "manywheel",
280280
"build_name": "manywheel-py3_13-cuda12_4",
281281
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",

‎tools/tests/assets/build_matrix_linux_wheel_nocpu.json

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
"gpu_arch_type": "cuda",
66
"gpu_arch_version": "11.8",
77
"desired_cuda": "cu118",
8-
"container_image": "pytorch/manylinux-builder:cuda11.8",
8+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
99
"package_type": "manywheel",
1010
"build_name": "manywheel-py3_9-cuda11_8",
1111
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -20,7 +20,7 @@
2020
"gpu_arch_type": "cuda",
2121
"gpu_arch_version": "12.4",
2222
"desired_cuda": "cu124",
23-
"container_image": "pytorch/manylinux-builder:cuda12.4",
23+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
2424
"package_type": "manywheel",
2525
"build_name": "manywheel-py3_9-cuda12_4",
2626
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -80,7 +80,7 @@
8080
"gpu_arch_type": "cuda",
8181
"gpu_arch_version": "11.8",
8282
"desired_cuda": "cu118",
83-
"container_image": "pytorch/manylinux-builder:cuda11.8",
83+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
8484
"package_type": "manywheel",
8585
"build_name": "manywheel-py3_10-cuda11_8",
8686
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -95,7 +95,7 @@
9595
"gpu_arch_type": "cuda",
9696
"gpu_arch_version": "12.4",
9797
"desired_cuda": "cu124",
98-
"container_image": "pytorch/manylinux-builder:cuda12.4",
98+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
9999
"package_type": "manywheel",
100100
"build_name": "manywheel-py3_10-cuda12_4",
101101
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -155,7 +155,7 @@
155155
"gpu_arch_type": "cuda",
156156
"gpu_arch_version": "11.8",
157157
"desired_cuda": "cu118",
158-
"container_image": "pytorch/manylinux-builder:cuda11.8",
158+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
159159
"package_type": "manywheel",
160160
"build_name": "manywheel-py3_11-cuda11_8",
161161
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -170,7 +170,7 @@
170170
"gpu_arch_type": "cuda",
171171
"gpu_arch_version": "12.4",
172172
"desired_cuda": "cu124",
173-
"container_image": "pytorch/manylinux-builder:cuda12.4",
173+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
174174
"package_type": "manywheel",
175175
"build_name": "manywheel-py3_11-cuda12_4",
176176
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -230,7 +230,7 @@
230230
"gpu_arch_type": "cuda",
231231
"gpu_arch_version": "11.8",
232232
"desired_cuda": "cu118",
233-
"container_image": "pytorch/manylinux-builder:cuda11.8",
233+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
234234
"package_type": "manywheel",
235235
"build_name": "manywheel-py3_12-cuda11_8",
236236
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -245,7 +245,7 @@
245245
"gpu_arch_type": "cuda",
246246
"gpu_arch_version": "12.4",
247247
"desired_cuda": "cu124",
248-
"container_image": "pytorch/manylinux-builder:cuda12.4",
248+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
249249
"package_type": "manywheel",
250250
"build_name": "manywheel-py3_12-cuda12_4",
251251
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -305,7 +305,7 @@
305305
"gpu_arch_type": "cuda",
306306
"gpu_arch_version": "11.8",
307307
"desired_cuda": "cu118",
308-
"container_image": "pytorch/manylinux-builder:cuda11.8",
308+
"container_image": "pytorch/manylinux2_28-builder:cuda11.8",
309309
"package_type": "manywheel",
310310
"build_name": "manywheel-py3_13-cuda11_8",
311311
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",
@@ -320,7 +320,7 @@
320320
"gpu_arch_type": "cuda",
321321
"gpu_arch_version": "12.4",
322322
"desired_cuda": "cu124",
323-
"container_image": "pytorch/manylinux-builder:cuda12.4",
323+
"container_image": "pytorch/manylinux2_28-builder:cuda12.4",
324324
"package_type": "manywheel",
325325
"build_name": "manywheel-py3_13-cuda12_4",
326326
"validation_runner": "linux.g5.4xlarge.nvidia.gpu",

0 commit comments

Comments
 (0)