错误描述
RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS
解决方案
添加如下代码
import torch
torch.backends.cudnn.benchmark = True
终于解决了,哭~
错误描述
RuntimeError: CuDNN error: CUDNN_STATUS_SUCCESS
解决方案
添加如下代码
import torch
torch.backends.cudnn.benchmark = True
终于解决了,哭~