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

[Easy] Bump CUDA nightly version to 11.8 / 12.4 / 12.6 in nightly pull tool (#143263)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143263
Approved by: https://github.com/malfet
This commit is contained in:
Xuehai Pan
2024-12-26 16:48:40 +08:00
committed by PyTorch MergeBot
parent 78502a58ba
commit 51a7ecde80
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tools/nightly.py
View File

@ -122,18 +122,18 @@ PIP_SOURCES = {
supported_platforms={"Linux", "Windows"},
accelerator="cuda",
),
"cuda-12.1": PipSource(
name="cuda-12.1",
index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/cu121",
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",
),
}