在代码中设置指定GPU，代码仍旧运行不成功

device = torch.device(“cuda:3” if torch.cuda.is_available() else “cpu”)
model = model.to(device)

在上面的代码前面添加了代码torch.cuda.set_device(2)设置当前设备。然后运行成功。

分析这可能是pytorch自身的原因。

RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR