RuntimeError: CUDA error: invalid device ordinal CUDA kernel errors might be asynchronously reported

---

问题描述：

在运行代码 args.device = torch.device('cuda: ' + str(args.gpu_id)) 设置GPU时出现如下问题 RuntimeError: CUDA error: invalid device ordinal CUDA kernel errors might be asynchronously reported

RuntimeError: CUDA error: 
invalid device ordinal CUDA kernel errors might be asynchronously reported at some other api call,
so the stacktrace below might be incorrect. 
for debugging cons