RuntimeError: cuDNN error: CUDNN_STATUS_INTERNAL_ERROR

在代碼中設置指定GPU，代碼仍舊運行不成功

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

在上面的代碼前面添加了代碼 torch.cuda.set_device(2)設置當前設備。然后運行成功。

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