测试densenet系列的模型，DenseBlock_layer.cu:231行总报错，提示 CUDNN_STATUS_NOT_INITIALIZED，不清楚原因。

senlinuc / caffe_ocr

主流ocr算法研究实验性的项目，目前实现了CNN+BLSTM+CTC架构

1.26k stars 535 forks source link

Closed hbulaoma closed 6 years ago

senlinuc commented 6 years ago

你用的是最新的代码吗？这两天DenseBlock_layer.cu我改了个batch size有变化时出错的bug。

hbulaoma commented 6 years ago

昨天晚上更新了最新的代码，仍然报错。inception-bn-res-blsm和resnet-res-blstm两套模型一直可以使用。不知道是显卡版本(GTX1080)的原因，还是代码造成的。

yaokeepmoving commented 6 years ago

最好将完整的报错信息粘贴或是截图出来，方便分析

hbulaoma commented 6 years ago

可以贴图吗？报错如下： F1113 20:15:52.651626 13012 DenseBlock_layer.cu:231] Check failed: status == CUDNN_STATUS_SUCCESS(1 vs. 0) CUDNN_STATUS_NOT_INITIZLIZED

senlinuc commented 6 years ago

是不是cudnn版本的问题？我的是7.0

prfans commented 6 years ago

这个可能是显卡驱动的问题,我是把cudnn7.0改为5.0解决的.