OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 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
Show Stats 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", accelerator="cpu",
), ),
# NOTE: Sync with CUDA_ARCHES in .github/scripts/generate_binary_build_matrix.py # 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( "cuda-12.6": PipSource(
name="cuda-12.6", name="cuda-12.6",
index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/cu126", index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/cu126",
supported_platforms={"Linux", "Windows"}, supported_platforms={"Linux", "Windows"},
accelerator="cuda", 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 # NOTE: Sync with ROCM_ARCHES in .github/scripts/generate_binary_build_matrix.py
"rocm-6.3": PipSource( "rocm-6.3": PipSource(
name="rocm-6.3", name="rocm-6.3",