【错误】
RuntimeError: CUDA error: invalid device ordinal
【解决方法】
args.device = torch.device('cuda:' + str(args.gpu_id))
使用的gpu_id超过装GPU的卡导致
我的代码是上面那种写法,不同代码写法不同,归根结底,把cuda:后面的数字改成合适的即可
【错误】
RuntimeError: CUDA error: invalid device ordinal
【解决方法】
args.device = torch.device('cuda:' + str(args.gpu_id))
使用的gpu_id超过装GPU的卡导致
我的代码是上面那种写法,不同代码写法不同,归根结底,把cuda:后面的数字改成合适的即可