loss.backward() cuDNN error: CUDNN

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版本进行下降。

我安装的是：

cu100/torch-1.1.0-cp36-cp36m-linux_x86_64.whl

然后就可以正常跑模型了