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

linux-rocm requirements don't pull torch-rocm #809

Closed
aaaaaaaaargh opened this issue Aug 12, 2024 · 1 comment · Fixed by #811
Closed

linux-rocm requirements don't pull torch-rocm #809

aaaaaaaaargh opened this issue Aug 12, 2024 · 1 comment · Fixed by #811
Labels
bug Something isn't working

Comments

@aaaaaaaaargh
Copy link

Description

As already mentioned in #803, manually installing the dependencies using the provided linux-rocm requirements file is not working as expected.

Steps to Reproduce

  1. Clone the repository in your scripts/addons folder as described in the documentation.
  2. Run /usr/share/blender/4.1/python/bin/python3.11 -m pip install -r requirements/linux-rocm.txt --target .python_dependencies in the checked out folder

Expected Behavior

A rocm version of PyTorch (>=2.3.1) should have been downloaded but instead it is downloading the default one withtout rocm support.

This is due to requirements/linux-rocm.txt referring to the extra-index-url https://download.pytorch.org/whl/nightly/rocm6.1/ which is only containing nightly versions and from my understanding the version constraint >=2.3.1 does not apply. Changing the extra-index-url to https://download.pytorch.org/whl/rocm6.1/ seems to fix the problem and I couldn't find any issues when using the addon in blender.

(Also, as a side note, the dropdown in the bug reporting form should be updated because the current gen AMD GPUs are supported now)

Addon Version

Other (Built from source)

GPU

AMD (unsupported for local generation)

@carson-katri
Copy link
Owner

I don't have a Linux machine to test with, can you confirm this PR looks correct?
#811

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
bug Something isn't working
Projects
None yet
Development

Successfully merging a pull request may close this issue.

2 participants