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)
这样就好了
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)
这样就好了