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

[Easy] update pip sources for CUDA in nightly pull tool (#149143)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/149143
Approved by: https://github.com/ezyang, https://github.com/cyyever
ghstack dependencies: #145685
This commit is contained in:
Xuehai Pan
2025-06-10 15:35:30 +08:00
committed by PyTorch MergeBot
parent 0319044e92
commit 2cdcd16e83
1 changed files with 6 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
tools/nightly.py
View File

@ -129,24 +129,18 @@ PIP_SOURCES = {
accelerator="cpu",
),
# NOTE: Sync with CUDA_ARCHES in .github/scripts/generate_binary_build_matrix.py
"cuda-11.8": PipSource(
name="cuda-11.8",
index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/cu118",
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",