pytorch cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

解决方法
1.测试

import torch
from torch.backends import cudnn
# 若正常则静默 
a = torch.tensor(1.)
# 若正常则静默 
a.cuda()
# 若正常则返回
tensor(1., device='cuda:0') 
 # 若正常则静默 
 cudnn.is_available()
 # 若正常则返回 True 
 cudnn.is_acceptable(a.cuda()) 
 # 若正常则返回 true

2.添加代码 在你的模型运行之前

A不在使用cudnn 
	torch.backends.cudnn.enabled = False
B坚持使用
	torch.backends.cudnn.enabled = True
	torch.backends.cudnn.benchmark = True

最终ok