【训练时如何指定GPU或者CPU】 【先赞后看养成习惯】求关注+点赞+收藏!!!😊 如果不知道自己电脑是否支持GPU或者CPU,我们可以使用下面这句命令,那么它会优先去调用GPU,如果没有GPU,就会去调用CPU device = torch.device("cuda" if torch.cuda.is_available() else "cpu") 如果要指定GPU,可以直接用下面的命令 device = torch.device("cuda") 如果要指定CPU,则用下面命令 device = torch.device("cpu")