如何判断这个代码是否使用gpu运行

可以通过以下代码来判断代码是否使用GPU运行：

import torch

if torch.cuda.is_available():
    device = torch.device("cuda")
    print("Using GPU")
else:
    device = torch.device("cpu")
    print("Using CPU")

在这段代码中，首先检查CUDA是否可用。如果CUDA可用，则将设备设置为GPU，否则将设备设置为CPU。如果输出结果为“Using GPU”，则说明代码在GPU上运行。