device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
device = torch.device(‘cuda‘ if torch.cuda.is_available() else ‘cpu‘)
最新推荐文章于 2024-06-22 17:22:33 发布
device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')