pytorch查看当前使用的是cpu还是gpu

当前使用的是CPU还是GPU可以通过以下代码查看：

import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(device)

如果输出结果为cuda，则当前使用的是GPU；如果输出结果为cpu，则当前使用的是CPU。