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] update pip sources for ROCm in nightly pull tool (#145685)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/145685
Approved by: https://github.com/ezyang
This commit is contained in:
Xuehai Pan
2025-06-10 15:35:29 +08:00
committed by PyTorch MergeBot
parent 9d2d227003
commit 0319044e92
2 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tools/nightly.py
View File

@ -128,6 +128,7 @@ PIP_SOURCES = {
supported_platforms={"Linux", "macOS", "Windows"},
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",
@ -146,6 +147,13 @@ PIP_SOURCES = {
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",
supported_platforms={"Linux"},
accelerator="rocm",
),
"rocm-6.4": PipSource(
name="rocm-6.4",
index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/rocm6.4",