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

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

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

• 给你找了一篇非常好的博客，你可以看看是否有帮助，链接：解决RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0])