Skip to content

ROCm/PyTorch install combinations#1

Merged
electron271 merged 2 commits into
electron271:mainfrom
sstamenk:rocm_install
Mar 9, 2026
Merged

ROCm/PyTorch install combinations#1
electron271 merged 2 commits into
electron271:mainfrom
sstamenk:rocm_install

Conversation

@sstamenk

@sstamenk sstamenk commented Mar 6, 2026

Copy link
Copy Markdown

Summary

  • Add new ROCm install extras in pyproject.toml for additional tested ROCm / PyTorch combinations:
    • rocm702-torch280 - Linux only
    • rocm711-torch291
    • rocm711-torch2100 - Linux only
    • rocm72-torch291
    • rocm72-torch2100 - Linux only
  • Remove rocm64-torch291 in favor of the newer explicit ROCm 7.x install matrix.
  • Restrict AMD bitsandbytes installs to Linux, since current bitsandbytes wheels are not available for Windows.

Validation

  • Tested the added combinations on:
    • AMD Instinct MI308X 192GB (gfx942)
    • Radeon RX 7900 XTX 24GB (gfx1100)
    • Radeon AI Pro R9700 32GB (gfx1201)
  • Windows was not tested.

Notes

  • rocm711-torch291 and rocm72-torch291 include Windows wheel entries, but I haven't validated those.

@electron271 electron271 merged commit 3d8f12b into electron271:main Mar 9, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants