怎么判断当前代码是否运行在gpu上

您可以使用以下代码来判断当前代码是否运行在GPU上：

import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print("当前代码运行在: ", device)

如果打印的结果为 "当前代码运行在: cuda" ，则说明当前代码在GPU上运行。如果打印的结果为 "当前代码运行在: cpu" ，则说明当前代码在CPU上运行。