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

Support for rocm amd gpu card #1271

Open

remixmabix opened this issue Jun 21, 2024 · 1 comment

remixmabix commented Jun 21, 2024

Pytorch is now very good with rocm how we can adapt to use rocm graphic cards with it ?

morti86 commented Jun 23, 2024

What OS are you running? As long as you get PyTorch with Rocm support it should be fine.

On Linux it works and I am running it on my 7900XT.
On Windows Pytorch should now be available with Rocm, but I am not sure.

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