diff --git a/tools/nightly.py b/tools/nightly.py index f1e64049727ed9..c5f5179ff3a15b 100755 --- a/tools/nightly.py +++ b/tools/nightly.py @@ -130,18 +130,18 @@ class PipSource(NamedTuple): supported_platforms={"Linux", "Windows"}, accelerator="cuda", ), - "cuda-12.4": PipSource( - name="cuda-12.4", - index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/cu124", - supported_platforms={"Linux", "Windows"}, - accelerator="cuda", - ), "cuda-12.6": PipSource( name="cuda-12.6", index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/cu126", supported_platforms={"Linux", "Windows"}, accelerator="cuda", ), + "cuda-12.8": PipSource( + name="cuda-12.8", + index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/cu128", + supported_platforms={"Linux", "Windows"}, + accelerator="cuda", + ), # NOTE: Sync with ROCM_ARCHES in .github/scripts/generate_binary_build_matrix.py "rocm-6.3": PipSource( name="rocm-6.3",