pytorch打印当前设备

A: 在PyTorch中，可以使用以下代码打印当前设备：

import torch

if torch.cuda.is_available():
    device = torch.device("cuda")  # GPU
    print("Current device: ", torch.cuda.get_device_name())
else:
    device = torch.device("cpu")
    print("Current device: CPU")

它将输出当前设备的名称，例如：`Current device: GeForce GTX 1080 Ti`。如果您的系统上没有可用的GPU，则将使用CPU作为默认设备。