Error with Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR

yjxiong / caffe

A fork of Caffe with OpenMPI-based Multi-GPU (mainly data parallel) support for action recognition and more. More documentation please see the original readme.

http://caffe.berkeleyvision.org/

Other

551 stars 154 forks source link

Error with Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR #174

Closed linquanxu closed 7 years ago

linquanxu commented 7 years ago

hi, There was an error happened on MobileNet(be similar with resnet50). qq 20170622110536

I guess it might be that the parameter group is too big. How to fix it? thanks.

yjxiong commented 7 years ago

I guess it's due to the large group number applied to cuDNN. Try using engine: CAFFE as a workaround in these layers.

linquanxu commented 7 years ago

ok，thanks for your response.