Getting Check failed: status == CUDNN_STATUS_SUCCESS (1 vs. 0) CUDNN_STATUS_NOT_INITIALIZED

[Coderx7]

When I want to use your prototxtfile for training, I get this error I have the latest version of caffe, cudnn v5.1 and cuda 8 installed! What is the cause for this and how can I get over it ? Thanks in advance

[rangerToby]

try to reduce the batch_size value in prototxt

[xiakj]

I met this problem when inference, like this:

I1016 10:26:46.874517 412 net.cpp:84] Creating Layer mobilenet0_conv21 I1016 10:26:46.875514 412 net.cpp:406] mobilenet0_conv21 <- mobilenet0_batchnorm20 I1016 10:26:46.875514 412 net.cpp:380] mobilenet0_conv21 -> mobilenet0_conv21 F1016 10:26:46.959211 412 cudnn_conv_layer.cpp:53] Check failed: status == CUDNN_STATUS_SUCCESS (1 vs. 0) CUDNN_STATUS_NOT_INITIALIZED *** Check failure stack trace: ***

how to solve it?

[WilliamWang1994]

@xiakj https://github.com/pby5/MobileNet_Caffe/issues/8