报错cuDNN: CUDNN_STATUS_EXECUTION_FAILED

加一句

torch.backends.cudnn.enabled = False

不行再查看cuda、python、pytorch、cudnn的版本兼容问题

原文地址:https://www.cnblogs.com/Manuel/p/12325409.html