RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED【解决方法】
RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED【解决方法】
在使用github上的开源框架pytorch-deeplab-xception在自己的数据集上进行语义分割时,遇到了这个错误。遇到坑debug了两天才解决。
按这个错误提示的意思搜了搜解决办法,说应该重装CuDNN、pytorch、显卡使得版本匹配才行。于是照做,折腾好久好久,发现没用。
因为这个错误出现在损失函数那里,于是又怀疑是不是输入维度的原因。于是把输入的label和Outpu
原创
2021-09-23 16:18:31 ·
13561 阅读 ·
3 评论