Confusing version numbers, nightlies, official releases, amd website, github readme, etc..

[peteralieber]

Many places say to do this:

pip install --pre torch torchvision torchaudio --index-url https://rocm.nightlies.amd.com/v2/gfx1151/

Great, but which rocm version do I install? My package says...

torch                                  2.11.0a0+rocm7.11.0a20260108
torchaudio                             2.11.0a0+rocm7.11.0a20260108
torchsde                               0.2.6
torchvision                            0.25.0a0+rocm7.11.0a20260108

Ok, so rocm 7.11?? The page here, https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/docs/install/installrad/windows/install-pytorch.html

Sure doesn't match, but when I installed, this command seems to like it:
python -m torch.utils.collect_env

But now I'm confused (yes, I know AI may not be right, but seriously, this seems just too hard to get into a power user mode...

I wonder if there is a place that all these versions, sdks, wheels, nightlies, releases, builds, etc. are more concretely defined and process described. This isn't even mentioning the ambiguity around therock and other builds of rocm. I want to be able to reason about these releases, what the requirements are to use a specific pytorch wheel, or how to get the right pytorch for a given therock build/release... Also, how to tell if a specific build is valid (i.e. the rocm build passed, but pytorch failed its CI/CD, etc.).

Anyone found any resources on this?

[marbre]

@peteralieber you're following the docs for ROCm 7.2 which is the current stable release. For the technology preview 7.11 see https://rocm.docs.amd.com/en/7.11.0-preview/index.html. However, if you want more bleeding edge packages, e.g. nightly builds based on TheRock, you're most likely looking for https://github.com/ROCm/TheRock/blob/main/RELEASES.md#installing-releases-using-pip.

If referring to TheRock produced releases we have

• "stable" releases (technology preview still, not officially stable train yet)
• prereleases
• nightly releases
• dev releases

See also https://github.com/ROCm/rockrel?tab=readme-ov-file#rockrel-therock-releases but please don't mix with docs for 7.2 :)

[peteralieber]

Okay so to be clear, if I want the latest pytorch build that passes tests and supports gfx 1151, My thought is I would need to go to the nightly builds, but just pulling the latest with pip from nightly's doesn't guarantee me to get a passing build right?

For this I would need to go to all the build reports and see which is the latest pie torch build for Windows. Specifically for this chip that passes the tests. Then I would need to check to see if the corresponding rocm build exists, passes tests.

If that pair doesn't exist then I would need to go back one build of the pytorch that passes tests and see if that corresponding rocm build works. This seems like a long process unless something I said here is a bad assumption 🤞🏼

[jammm]

If you want the latest pytorch wheels that pass tests, the safe choice is to go for the "Stable" releases i.e., ROCm 7.2 ones from https://rocm.docs.amd.com/projects/radeon-ryzen/en/latest/docs/install/installrad/windows/install-pytorch.html.

The nightlies, dev and pre-releases are likely not what you're looking for if you want a stable experience. If however you're willing to experiment and handle potential instability, then you can go ahead with the nightly wheels which you were already using with pip install --pre torch torchvision torchaudio --index-url https://rocm.nightlies.amd.com/v2/gfx1151/.

[ScottTodd]

Docs: https://github.com/ROCm/TheRock/blob/main/docs/packaging/versioning.md