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

[ROCm] upgrade nightly wheels to rocm6.4 (#151355)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/151355
Approved by: https://github.com/jeffdaily

Co-authored-by: Jeff Daily <jeff.daily@amd.com>
This commit is contained in:
Jithun Nair
2025-04-17 17:29:07 +00:00
committed by PyTorch MergeBot
parent ef64beb232
commit b4550541ea
9 changed files with 653 additions and 655 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tools/nightly.py
View File

@ -141,9 +141,9 @@ PIP_SOURCES = {
supported_platforms={"Linux", "Windows"},
accelerator="cuda",
),
"rocm-6.2.4": PipSource(
name="rocm-6.2.4",
index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/rocm6.2.4",
"rocm-6.4": PipSource(
name="rocm-6.4",
index_url=f"{PYTORCH_NIGHTLY_PIP_INDEX_URL}/rocm6.4",
supported_platforms={"Linux"},
accelerator="rocm",
),