本文主要是介绍ubuntu 下出现 cuDNN Error: CUDNN_STATUS_OUT_INITIALIZED错误,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
问题描述:换了一个模型跑代码,就出现了标题的问题
问题解决:可以看到出错之前是在loss.backward这个地方出错的,所以回去检查模型最后的输出outputs和标签labels获得的loss,发现在计算loss的时候用的是不带softmax的BCELoss() 函数,模型最后的输出也没有另外添加softmax()函数。所以出错了。
解决方案:1. 将outputs先输入softmax函数在进行criterion
2. 用自带softmax的损失函数BCEWithLogitsLoss作为criterion
这篇关于ubuntu 下出现 cuDNN Error: CUDNN_STATUS_OUT_INITIALIZED错误的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!