报错:
解决RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
原因:
1.可能cuda与python版本不一致
详见:https://blog.csdn.net/qq_39938666/article/details/86611474
2.可能是你的GPU显存被占用光了,再运行就出错了。
我的用cpu,代码就能跑了
device = torch.device("cuda")
改为
device = torch.device("cpu")