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