-
Notifications
You must be signed in to change notification settings - Fork 121
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
Amd gpu supports #20
Comments
Makes sense! |
I believe it's should be straightforward just have to install rocm hips library and using pytorch index url pointing to rocm5.x . I have a smaller fedora docker for rocm pytorch, because the official amd pytorch one is bloated AF. Almost 35GB . Mine as a base is 🫠only ~10GB.hopefully i will have time in the week coming to put together a draft PR for the Dockerfile change |
Should be fairly easy with HIP |
We have this PR: that we'll test somehow (don't have any AMD GPUs), or at least we'll set up an auto docker build and someone will test it :D |
In the coming days i will definitely test it. 👍👍 i have both motivation and low end amdgpu hardware 🙃 |
Any progress on this matter? I also have an AMD GPU. |
hi, is there any plan to support amd gpus or even any alternatives?
amd gpus have excellent vram capacity and power with reasonable price
The text was updated successfully, but these errors were encountered: