【训练时如何指定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")