Pytorch training with GPU on Mac

The code driving the GPU in the Mac should be changed to:

device = torch.device("mps")

Guess you like

Origin blog.csdn.net/qq_45191169/article/details/128211090