原文:报错cuDNN: CUDNN_STATUS_EXECUTION_FAILED

加一句 torch.backends.cudnn.enabled False 不行再查看cuda python pytorch cudnn的版本兼容问题 ...

2020-02-18 11:33 0 1386 推荐指数:

查看详情

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED 解决方法

最近再用LSTM训练模型,有时候epoch进行几次后总会出现RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED,心态都快崩了。记录一下自己踩过的坑吧,上网查了一下这个报错,大家说法不一,有人说是因为pytorch和cuda版本不对应 ...

Sun Jan 10 05:49:00 CST 2021 0 10605
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM