报错RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED解决方法

报错RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

CUDA 9.0 cudnn=7.6.5 运行代码报错：

这是由于CUDA版本为9.0的原因，在train代码中加入下面代码即可正常运行：

// 加入train.py中
torch.backends.cudnn.benchmark = True;