Skip to content

Conversation

@B4rr3l-Rid3r
Copy link
Contributor

Updated requirements for ROCm and Torch

Updated requirements for ROCm and Torch
@carson-katri carson-katri merged commit e0b07e0 into carson-katri:releases/0.4.0 Jul 27, 2024
@carson-katri
Copy link
Owner

Thanks!

@aaaaaaaaargh
Copy link

aaaaaaaaargh commented Aug 11, 2024

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

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.

3 participants