Update ROCM version to 7.1 from 6.3 #8867
Open
+1,949
−1,920
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
- Update PyTorch to 2.10.0+rocm7.1 (from 2.7.1+rocm6.3) - Update TorchVision to 0.25.0+rocm7.1 (from 0.22.1+rocm6.3) - Update PyTorch ROCm index URL to rocm7.1 - Update Docker ROCm runtime to 7.1.1 (from 6.3.4) ROCm 7.1 adds support for 3 new GPU architectures: - gfx950: CDNA3+ (Instinct MI350, MI325X) - gfx1150: RDNA 3.5 (Strix Point APU) - gfx1151: RDNA 3.5 (Strix Halo APU) All 11 previously supported architectures remain compatible. Total supported architectures: 14 (11 existing + 3 new) Relates to invoke-ai#8655
The uv package manager couldn't resolve triton-rocm as a transitive dependency of torch from the PyTorch custom index. This commit adds triton-rocm==3.6.0 explicitly to the rocm extras with a Linux-only platform marker. Changes: - Added triton-rocm==3.6.0; sys_platform == 'linux' to rocm extras - Added triton-rocm source configuration pointing to torch-rocm index - Already committed: torch version constraint relaxation (>=2.7.0,<3.0) - Already committed: platform environment restrictions for ROCm Verified with uv sync --extra rocm: - torch==2.10.0+rocm7.1 installed correctly - torchvision==0.25.0+rocm7.1 installed correctly - triton-rocm==3.6.0 resolved successfully - ROCm 7.1.25424 detected on AMD Radeon PRO V620 - All 8 model architectures accessible - GPU compute operations verified Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
Changed torch backend from rocm6.3 to rocm7.1 in manual installation instructions for Invoke v5.12 and later. Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
github-actions
bot
added
docker
Root
docs
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
1 participant
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Summary
Update ROCM version to 7.1 from 6.3
Related Issues / Discussions
Part 1 of issue #8655
QA Instructions
Only affects ROCM users, just need to make sure it runs.
Checklist
Dockers build: https://github.com/heathen711/InvokeAI/actions/runs/21847900570