RuntimeError: Invalid device string: 'cuda:-1'

Question

device = torch.device("cuda:{}".format(args.gpu) if torch.cuda.is_available() else "cpu")
RuntimeError: Invalid device string: 'cuda:-1'
请问这种情况如何解决？

Answer 1

试下改成
device = torch.device('cuda' if torch.cuda.is_available() and use_cuda else 'cpu')