pytorch wheels for gfx1151
Pre-release
Pre-release
Self-contained Pytorch wheel for the Strix Halo (gfx1151):
- Windows: torch-2.7.0a0+git3f903c3-cp312-cp312-win_amd64.whl, torchvision-0.22.0+9eb57cd-cp312-cp312-win_amd64.whl, torchaudio-2.6.0a0+1a8f621-cp312-cp312-win_amd64.whl
- Linux: torch-2.7.0a0+gitbfd8155-cp311-cp311-linux_x86_64.whl
Download and install with pip install <WHEEL_FILE>.
Contents:
- Strix Halo (gfx1151) GPU support
- Pytorch 2.7
- ROCm 6.5.0rc
- Built for Python 3.11
The Windows wheel was built by @jammm, the Linux wheel by @scottt
- Send feedback here
- This is a developer build released by outside contributors to the ROCm project
Update:
- Replaced S3 with Github links after @jammm recompressed the wheels to fit under Github's 2GB limit