@@ -12,7 +12,7 @@ permissions:
1212
1313jobs :
1414 build :
15- name : forge-cu129 -nightly
15+ name : forge-cu128 -nightly
1616 uses : pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
1717 strategy :
1818 fail-fast : false
@@ -31,13 +31,13 @@ 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",
40- "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu129 ",
40+ "installation": "pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu128 ",
4141 "channel": "nightly",
4242 "upload_to_base_bucket": "no",
4343 "stable_version": "2.8.0",
0 commit comments