Pytorch用Mac上的GPU训练

Mac中驱动GPU的代码应改为:

device = torch.device("mps")

猜你喜欢

转载自blog.csdn.net/qq_45191169/article/details/128211090