> 文档中心 > 判断torch.device是gpu还是cpu

判断torch.device是gpu还是cpu


判断torch.device是gpu还是cpu

这样直接判断是有问题会有问题,因为torch.device是class类型

 device=torch.device('cuda:0') if device!= "cpu":    print("GPU,&&&&") else:    print("CPU,&&&&")

需要改为:

 device=torch.device('cuda:0') if device.type != "cpu":    print("GPU,&&&&") else:    print("CPU,&&&&")

医学名词百科