pytorch 测试gpu可用

您可以使用以下代码测试PyTorch是否可以访问GPU：

import torch

if torch.cuda.is_available():
    device = torch.device("cuda")
    print("GPU可用")
else:
    device = torch.device("cpu")
    print("GPU不可用")

print("当前设备：", device)

这段代码首先检查PyTorch是否可以访问GPU，如果可用，则将设备设置为cuda。否则，将设备设置为cpu。最后，它会打印出当前使用的设备。