AMD GPUs #18763

fses91 started this conversation in General

fses91
Oct 9, 2023

Hi,
does PytorchLightning also work with AMD GPUs? Because as I saw there is no accelerator for AMD GPUs https://github.com/Lightning-AI/lightning/tree/master/src/lightning/pytorch/accelerators

Or does it work with the CUDA accelerators?

Best regards

Replies: 0 comments

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