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.

Sign up for GitHub

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

Jump to bottom

Check AMD GPU support #128

Open

jatkinson1000 opened this issue May 13, 2024 · 1 comment

Assignees

Member

jatkinson1000 commented May 13, 2024 •

edited

Loading

@TomMelt suggested we check support on AMD GPU devices and offered to test on a suitable machine.

To do this may require modifications akin to those in #125.

jatkinson1000 assigned TomMelt

jatkinson1000 changed the title ~~Check ARM GPU support~~ Check AMD GPU support

Member Author

jatkinson1000 commented Jun 3, 2024

I think this would require downloading ROCM compatible rather than CUDA compatible libtorch as described here: https://pytorch.org/blog/pytorch-for-amd-rocm-platform-now-available-as-python-package/

We would then need to add the relevant enum to target the AMD backend.

This is perhaps something that could be combined in #125, or at least follow a very similar pattern.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment