I’m writing code for a Quantum GAN, using a Mac M1. I can’t seem to find any support for Mac GPUs, in order to link those with PyTorch. Is there any support for it?
Hi @A9B8C7D6
To make use of PyTorch with M1 GPUs I suggest following the notes from Introducing Accelerated PyTorch Training on Mac | PyTorch
Though, while it may be possible to install a working PyTorch with Apple Metal Performance Shader (MPS) support, we currently do not officially support this through PennyLane. Though, please feel free to let us know if you succeed in getting this to work. Hopefully it is possible to simply swap “cuda” to “mps” for Tensors to get a working example.
1 Like