报错:
line 93, in backward allow_unreachable=True) # allow_unreachable flag RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
背景:
我这里是处在loss函数计算的部分报错了。
先将device切换至cpu运行看是否报错,如果也出错,大概率是代码本身编写问题。
如果cpu没有报错,建议最好将torch版本进行下降。
我是使用的torch1.2.0的torch
NVIDIA-SMI 410.93 Driver Version: 410.93 CUDA Version: 10.0
>>> torch.version.cuda
'10.0.130'
解决方案:
将torch版本进行下降。
我安装的是: