Skip to content

Update ROCM version to 7.1 from 6.3 (resubmission) #9296

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
lstein wants to merge 9 commits into from
Closed

Update ROCM version to 7.1 from 6.3 (resubmission) #9296

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
60450b7
Upgrade ROCm support from 6.3 to 7.1
heathen711 Feb 9, 2026
3ba1196
Fix uv sync for ROCm 7.1 by explicitly declaring triton-rocm
heathen711 Feb 9, 2026
776cd58
Update uv lock and remove old dockerfile
heathen711 Feb 10, 2026
45a60ba
Update installation docs for ROCm 7.1
heathen711 Feb 10, 2026
ed54fde
Merge main into enhancement/heathen711/issue_8655
May 13, 2026
14a9c99
docs: update ROCm torch backend from 6.3 to 7.1
May 13, 2026
24672e4
fix: regenerate uv.lock with ROCm 7.1 dependencies
May 13, 2026
42dae86
Merge branch 'main' into enhancement/heathen711/issue_8655
joshistoast May 15, 2026
d0f5a01
remove redundant inclusion of torch-triton wheel && confirm no aarch6…
lstein Jun 23, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
136 changes: 0 additions & 136 deletions docker/Dockerfile-rocm-full
View file
Open in desktop

This file was deleted.

2 changes: 1 addition & 1 deletion docs-old/installation/manual.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,7 @@ The following commands vary depending on the version of Invoke being installed a

- If you are on Windows or Linux with an Nvidia GPU, use `--torch-backend=cu128`.
- If you are on Linux with no GPU, use `--torch-backend=cpu`.
- If you are on Linux with an AMD GPU, use `--torch-backend=rocm6.3`.
- If you are on Linux with an AMD GPU, use `--torch-backend=rocm7.1`.
- **In all other cases, do not use a torch backend.**

=== "Invoke v5.10.0 to v5.11.0"
Expand Down
2 changes: 1 addition & 1 deletion docs/src/content/docs/start-here/manual.mdx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,7 @@ The following commands vary depending on the version of Invoke being installed a
<TabItem label="Linux + AMD GPU">
Use:
```sh
--torch-backend=rocm6.3
--torch-backend=rocm7.1
```
</TabItem>
<TabItem label="All other cases">
Expand Down
17 changes: 10 additions & 7 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,7 @@ dependencies = [
"safetensors",
"sentencepiece==0.2.0", # 0.2.1 coredumps windows when loading t5 tokenizer
"spandrel",
"torch~=2.7.0", # torch and related dependencies are loosely pinned, will respect requirement of `diffusers[torch]`
"torch>=2.7.0,<3.0", # torch and related dependencies are loosely pinned, will respect requirement of `diffusers[torch]`
"torchsde", # diffusers needs this for SDE solvers, but it is not an explicit dep of diffusers
"torchvision",
"transformers>=4.56.0",
Expand Down Expand Up @@ -93,9 +93,9 @@ dependencies = [
"cpu" = ["torch==2.7.1+cpu", "torchvision==0.22.1+cpu"]
"cuda" = ["torch==2.7.1+cu128", "torchvision==0.22.1+cu128"]
"rocm" = [
"torch==2.7.1+rocm6.3",
"torchvision==0.22.1+rocm6.3",
"pytorch-triton-rocm",
"torch==2.10.0+rocm7.1",
"torchvision==0.25.0+rocm7.1",
"triton-rocm==3.6.0; sys_platform == 'linux'",
]

"onnx" = ["onnxruntime"]
Expand Down Expand Up @@ -130,6 +130,9 @@ dependencies = [
override-dependencies = ["opencv-python; sys_platform=='never'"]
conflicts = [[{ extra = "cpu" }, { extra = "cuda" }, { extra = "rocm" }]]
index-strategy = "unsafe-best-match"
# We do not publish or support ARM Linux builds; restrict resolution to
# x86_64 Linux, Windows, and macOS. (ROCm in particular is x86_64-only.)
environments = ["sys_platform == 'win32' or sys_platform == 'darwin' or (sys_platform == 'linux' and platform_machine == 'x86_64')"]

[tool.uv.sources]
torch = [
Expand All @@ -142,8 +145,8 @@ torchvision = [
{ index = "torch-cuda", extra = "cuda" },
{ index = "torch-rocm", extra = "rocm" },
]
pytorch-triton-rocm = [
{ index = "torch-rocm", extra = "rocm", marker = "sys_platform == 'linux'" },
triton-rocm = [
{ index = "torch-rocm", marker = "sys_platform == 'linux'" },
]

[[tool.uv.index]]
Expand All @@ -158,7 +161,7 @@ explicit = true

[[tool.uv.index]]
name = "torch-rocm"
url = "https://download.pytorch.org/whl/rocm6.3"
url = "https://download.pytorch.org/whl/rocm7.1"
explicit = true

[project.scripts]
Expand Down
Loading