device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
一句话判断GPU是否可用 ,否则使用cpu
最新推荐文章于 2023-01-01 20:23:24 发布
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")