如何测试pytorch可以用gpu

来加速模型训练？

你可以使用以下代码来测试pytorch是否可以使用GPU来加速模型训练：

import torch

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

x = torch.randn(10, 10).to(device)
y = torch.randn(10, 10).to(device)

# 在GPU上使用矩阵乘法
z = torch.mm(x, y)

如果输出的结果在GPU上运行，则说明PyTorch可以使用GPU来加速模型训练。