From eca6ac2293178a7d0fe77216eb882117d6aa0f7e Mon Sep 17 00:00:00 2001 From: Xuehai Pan Date: Tue, 30 Sep 2025 14:56:06 +0800 Subject: [PATCH] [BE][Easy] update CUDA and ROCm sources in nightly tool (#162324) Pull Request resolved: https://github.com/pytorch/pytorch/pull/162324 Approved by: https://github.com/ezyang --- tools/nightly.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tools/nightly.py b/tools/nightly.py index d3d9b4e77b51..e8d4eef8004f 100755 --- a/tools/nightly.py +++ b/tools/nightly.py @@ -142,10 +142,16 @@ PIP_SOURCES = { supported_platforms={"Linux", "Windows"}, accelerator="cuda", ), + "cuda-13.0": PipSource( + name="cuda-13.0", + index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/cu130", + 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", - index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/rocm6.3", + "rocm-6.4": PipSource( + name="rocm-6.4", + index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/rocm6.4", supported_platforms={"Linux"}, accelerator="rocm", ),