判断torch.device是gpu还是cpu

Posted 2022-10-01 AI浩

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了判断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,&&&&")

以上是关于判断torch.device是gpu还是cpu的主要内容，如果未能解决你的问题，请参考以下文章