报错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;
报错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;