The GPU on Mac is not Nvidia’s kind. It is M1 GPU designed by Apple.
You need to first set the device to mps.
See this for more details on installation and usage:
Running PyTorch on the M1 GPUtype or paste code here
3 Likes