device = torch.device("cuda: 0" if torch.cuda.is_available() else "cpu")
print(device)
尝试把中间的空格去了 我就是的
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)
这样就好了
这段代码根据CUDA是否可用,自动将设备设置为GPU(cuda:0)或CPU。如果torch.cuda.is_available()返回True,设备将是GPU,否则默认为CPU。
device = torch.device("cuda: 0" if torch.cuda.is_available() else "cpu")
print(device)
尝试把中间的空格去了 我就是的
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device)
这样就好了
1758
3388
6644
417
417
308
318
314

被折叠的 条评论
为什么被折叠?
