原文:報錯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