torch的torch device apple m1 使用哪种比较好请给出代码

在Apple M1上，可以使用torch.device('cpu')来使用CPU运算，也可以使用torch.device('cuda')来使用GPU运算，但需要安装适用于M1芯片的cuda版本。

如果要在M1上使用GPU加速，可以使用Metal作为后端来运行PyTorch，代码如下：

import torch

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

if device.type == 'cuda':
    torch.backends.metal.enabled = True

print('Using device:', device)

这段代码会检查是否有CUDA可用，如果有，则启用Metal后端，如果没有，则使用CPU运算。