m系列芯片使用加速 CUDA代码 USE_CUDA = torch.cuda.is_available() if USE_CUDA: model = model.cuda() MPS代码 USE_MPS = torch.backends.mps.is_available() if USE_MPS: mps_device = torch.device("mps") model = model.to(mps_device)