参考文章 解决RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
pytorch torch.backends.cudnn设置作用 一开始我以为是数据量太大的问题，但是我用4张显卡跑也会报错，看了看issue都说减少batch size，但是对我没用，查了上面的参考文章，加了以下代码，发现有用。 加以下代码 torch.backends.cudnn.enabled = False
