RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

 

 参考：（已解决）pytorch错误：RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED （安装cuda）_OpenSceneGraph的博客-程序员宝宝 - 程序员宝宝 (cxybb.com)

 

import torch
print(torch.__version__)
print(torch.version.cuda) 查看torch版本支持的cuda版本

根据上文提示，我换了一台cuda版本为10.2的服务器，慢慢悠悠的开始跑。

第二天我开始继续开始想问题。
Getting error, RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED while running a basic RNN model - Deep Learning (Training & Inference) / TensorRT - NVIDIA Developer Forums
指向下文
Cuda Error : RuntimeError: CUDNN_STATUS_EXECUTION_FAILED - PyTorch Forums
根据论坛中的提问和回答，猜测是因为一些张量没有cuda到GPU中计算导致的。