Aaryaman “Jam” Vasishta
@adyaman.bsky.social
Former ML+3D Engineer @ Stability AI
Ex. AMD Research Engineer, RT & Neural Rendering
2021 Graduate, Computer Graphics Group @ University of Tokyo.
https://aaryaman.net
Ex. AMD Research Engineer, RT & Neural Rendering
2021 Graduate, Computer Graphics Group @ University of Tokyo.
https://aaryaman.net
This is a "fat" wheel which bundles ROCm binaries alongside pytorch, so you won't need to download/install ROCm yourself.
You can get even higher performance when using “—gpu-only” - give it a try!
You can get even higher performance when using “—gpu-only” - give it a try!
May 30, 2025 at 6:12 AM
This is a "fat" wheel which bundles ROCm binaries alongside pytorch, so you won't need to download/install ROCm yourself.
You can get even higher performance when using “—gpu-only” - give it a try!
You can get even higher performance when using “—gpu-only” - give it a try!
Python312 Wheel can be downloaded from
github.com/scottt/rocm-...
Currently tested to work fast for gfx1151 but It should technically support gfx1100,gfx110{1,2,3}, gfx1151 gfx1201, gfx1151. Basically 7900(XT)/7800(XT)/7600(XT)/phoenix, 9070(XT) (navi48) and and strix halo (8060s).
github.com/scottt/rocm-...
Currently tested to work fast for gfx1151 but It should technically support gfx1100,gfx110{1,2,3}, gfx1151 gfx1201, gfx1151. Basically 7900(XT)/7800(XT)/7600(XT)/phoenix, 9070(XT) (navi48) and and strix halo (8060s).
Release Pytorch wheels for gfx110x, gfx1151, and gfx1201 · scottt/rocm-TheRock
Self-contained Pytorch wheels for gfx1100, gfx1101, gfx1102, gfx1103, gfx1151, and gfx1201 GPUs for Python 3.12 on Windows by @jammm
Mainly tested and known to run fast on the Strix Halo (gfx1151)...
github.com
May 30, 2025 at 6:12 AM
Python312 Wheel can be downloaded from
github.com/scottt/rocm-...
Currently tested to work fast for gfx1151 but It should technically support gfx1100,gfx110{1,2,3}, gfx1151 gfx1201, gfx1151. Basically 7900(XT)/7800(XT)/7600(XT)/phoenix, 9070(XT) (navi48) and and strix halo (8060s).
github.com/scottt/rocm-...
Currently tested to work fast for gfx1151 but It should technically support gfx1100,gfx110{1,2,3}, gfx1151 gfx1201, gfx1151. Basically 7900(XT)/7800(XT)/7600(XT)/phoenix, 9070(XT) (navi48) and and strix halo (8060s).