frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE][Easy] update CUDA and ROCm sources in nightly tool (#162324)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/162324
Approved by: https://github.com/ezyang
This commit is contained in:
Xuehai Pan
2025-09-30 14:56:06 +08:00
committed by PyTorch MergeBot
parent 12d4cb0122
commit eca6ac2293
1 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tools/nightly.py
View File

@ -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",
),