答案网_找答案上万事OK问答库

问题:cuDNN error:CUDNN_STATUS_EXECUTION_FAILED

【答案】:cuDNN error:CUDNN_STATUS_EXECUTION_FAILED

在训练RNN时很容易出现这个问题,cudnn其实就是cuda的一个工具,专门用于深度网络加速,当然不用cudnn也是可以训练的,只是训练速度会慢,cuda也可以使用其他工具进行加速,但是效果不会这么好。

我是在使用RNN时遇见了这个问题,使用的是LSTM,

在这个网站上找到了解决方案
Cuda Error : RuntimeError: CUDNN_STATUS_EXECUTION_FAILED​
discuss.pytorch.org/t/cuda-error-runtimeerror-cudnn-status-execution-failed/17625/10

这个人的方法:减小batch_size,

还有一种可能会遇到这样的问题,就是随着模型的训练,一直有数据进入显存,就是把数据放进了显存中,也会出现这样的错误,检查生成数据的代码。

还有一种一定可以的方法那就是禁用cudnn

torch.backends.cudnn.enabled=False


DataLoader中num_workers参数的设置,如果在Windows上设置num_workers为大于1的数,则将其改为默认值0,函数内部会为num_workers设置一个默认值;
建议和主机的cpu数量相同。

 


【关键字】:

首页 > IT » cuDNN error:CUDNN_STATUS_EXECUTION_FAILED

相关推荐

首页 |  家庭 | 升学 | 小学 | 中考 | 高考 | 考研 | 养老 | 育儿 | 旅游 | 宠物 | 美丽 | 家居 | 兴趣 | 健康 | 美食 | 商业 | 交通 | 艺术 | IT | 教育 | 体育 | 
© 2020 答案网_找答案上万事OK问答库