在代码中设置指定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
原文：https://www.cnblogs.com/leifzhang/p/15270244.html