Skip to content

Commit d9e9aa8

Browse files
committed
Build Monarch with CUDA 12.8 instead of 12.9
1 parent 29a7d50 commit d9e9aa8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

.github/workflows/build_wheels.yaml

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -31,11 +31,11 @@ jobs:
3131
{
3232
"python_version": "3.10",
3333
"gpu_arch_type": "cuda",
34-
"gpu_arch_version": "12.9",
35-
"desired_cuda": "cu129",
36-
"container_image": "pytorch/manylinux2_28-builder:cuda12.9",
34+
"gpu_arch_version": "12.8",
35+
"desired_cuda": "cu128",
36+
"container_image": "pytorch/manylinux2_28-builder:cuda12.8",
3737
"package_type": "manywheel",
38-
"build_name": "manywheel-py3_10-cuda12_9",
38+
"build_name": "manywheel-py3_10-cuda12_8",
3939
"validation_runner": "linux.4xlarge.nvidia.gpu",
4040
"installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129",
4141
"channel": "nightly",

0 commit comments

Comments
 (0)