Closed B4rr3l-Rid3r closed 1 month ago
Thanks!
Quick question about this PR. How is this supposed to work? As far as I can see there is no stable version of Torch that fulfills this requirement, so 2.4.0 stable non-rocm version is being downloaded instead.
In order to get it right I had do change linux-rocm.txt
like this:
--extra-index-url https://download.pytorch.org/whl/nightly/rocm6.1/
torch==2.5.0.dev20240811+rocm6.1
Obviously not a very elegant solution since it requires the explicit nightly version tags, but maybe I'm missing something here.
EDIT: Another more approachable way would be not to use nightly at all.
--extra-index-url https://download.pytorch.org/whl/rocm6.1/
torch>=2.3.1
Updated requirements for ROCm and Torch