Skip to content

Commit 0e0b3a9

Browse files
committed
Add Windows 13.0 build
1 parent ba6d82f commit 0e0b3a9

2 files changed

Lines changed: 75 additions & 3 deletions

File tree

tools/scripts/generate_binary_build_matrix.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -445,9 +445,6 @@ def generate_wheels_matrix(
445445
upload_to_base_bucket = "no"
446446
if os in (LINUX, WINDOWS):
447447
arches += CUDA_ARCHES
448-
# todo: remove once windows cuda 13.0 binaries are available
449-
if channel == NIGHTLY and os != LINUX:
450-
arches.remove("13.0")
451448
elif os == LINUX_AARCH64:
452449
arches += CUDA_AARCH64_ARCHES
453450

tools/tests/assets/build_matrix_windows_wheel_cuda.json

Lines changed: 75 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,21 @@
4545
"stable_version": "2.8.0",
4646
"use_split_build": false
4747
},
48+
{
49+
"python_version": "3.10",
50+
"gpu_arch_type": "cuda",
51+
"gpu_arch_version": "13.0",
52+
"desired_cuda": "cu130",
53+
"container_image": "pytorch/manylinux2_28-builder:cuda13.0",
54+
"package_type": "wheel",
55+
"build_name": "wheel-py3_10-cuda13_0",
56+
"validation_runner": "windows.g4dn.xlarge",
57+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130",
58+
"channel": "nightly",
59+
"upload_to_base_bucket": "no",
60+
"stable_version": "2.8.0",
61+
"use_split_build": false
62+
},
4863
{
4964
"python_version": "3.10",
5065
"gpu_arch_type": "xpu",
@@ -105,6 +120,21 @@
105120
"stable_version": "2.8.0",
106121
"use_split_build": false
107122
},
123+
{
124+
"python_version": "3.11",
125+
"gpu_arch_type": "cuda",
126+
"gpu_arch_version": "13.0",
127+
"desired_cuda": "cu130",
128+
"container_image": "pytorch/manylinux2_28-builder:cuda13.0",
129+
"package_type": "wheel",
130+
"build_name": "wheel-py3_11-cuda13_0",
131+
"validation_runner": "windows.g4dn.xlarge",
132+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130",
133+
"channel": "nightly",
134+
"upload_to_base_bucket": "no",
135+
"stable_version": "2.8.0",
136+
"use_split_build": false
137+
},
108138
{
109139
"python_version": "3.11",
110140
"gpu_arch_type": "xpu",
@@ -165,6 +195,21 @@
165195
"stable_version": "2.8.0",
166196
"use_split_build": false
167197
},
198+
{
199+
"python_version": "3.12",
200+
"gpu_arch_type": "cuda",
201+
"gpu_arch_version": "13.0",
202+
"desired_cuda": "cu130",
203+
"container_image": "pytorch/manylinux2_28-builder:cuda13.0",
204+
"package_type": "wheel",
205+
"build_name": "wheel-py3_12-cuda13_0",
206+
"validation_runner": "windows.g4dn.xlarge",
207+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130",
208+
"channel": "nightly",
209+
"upload_to_base_bucket": "no",
210+
"stable_version": "2.8.0",
211+
"use_split_build": false
212+
},
168213
{
169214
"python_version": "3.12",
170215
"gpu_arch_type": "xpu",
@@ -225,6 +270,21 @@
225270
"stable_version": "2.8.0",
226271
"use_split_build": false
227272
},
273+
{
274+
"python_version": "3.13",
275+
"gpu_arch_type": "cuda",
276+
"gpu_arch_version": "13.0",
277+
"desired_cuda": "cu130",
278+
"container_image": "pytorch/manylinux2_28-builder:cuda13.0",
279+
"package_type": "wheel",
280+
"build_name": "wheel-py3_13-cuda13_0",
281+
"validation_runner": "windows.g4dn.xlarge",
282+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130",
283+
"channel": "nightly",
284+
"upload_to_base_bucket": "no",
285+
"stable_version": "2.8.0",
286+
"use_split_build": false
287+
},
228288
{
229289
"python_version": "3.13",
230290
"gpu_arch_type": "xpu",
@@ -285,6 +345,21 @@
285345
"stable_version": "2.8.0",
286346
"use_split_build": false
287347
},
348+
{
349+
"python_version": "3.13t",
350+
"gpu_arch_type": "cuda",
351+
"gpu_arch_version": "13.0",
352+
"desired_cuda": "cu130",
353+
"container_image": "pytorch/manylinux2_28-builder:cuda13.0",
354+
"package_type": "wheel",
355+
"build_name": "wheel-py3_13t-cuda13_0",
356+
"validation_runner": "windows.g4dn.xlarge",
357+
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130",
358+
"channel": "nightly",
359+
"upload_to_base_bucket": "no",
360+
"stable_version": "2.8.0",
361+
"use_split_build": false
362+
},
288363
{
289364
"python_version": "3.13t",
290365
"gpu_arch_type": "xpu",

0 commit comments

Comments
 (0)