检验gpu是否可用的pytorch代码

要检查GPU是否可用的PyTorch代码如下所示：

import torch

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

print(device)

这段代码会检查是否有可用的CUDA设备，如果有则将设备设置为GPU并输出。否则将设备设置为CPU并输出。