Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update linux-rocm.txt #803

Merged
merged 1 commit into from
Jul 27, 2024

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