m1 pro在pytorch怎么写调用gpu语句

在 PyTorch 中调用 GPU 的语句是使用 `cuda()` 方法将数据移动到 GPU 上。例如：

import torch

# 创建一个 tensor
x = torch.randn(10)

# 将 tensor 移动到 GPU 上
x = x.cuda()

# 在 GPU 上进行运算
y = x * 2

# 将 tensor 从 GPU 上移回 CPU
y = y.cpu()

注意，如果你的电脑没有 GPU，或者没有安装 CUDA，那么调用 `cuda()` 方法会报错。在这种情况下，可以使用 `cpu()` 方法将 tensor 移回 CPU 上。