HsinYingLee / OPN

46 stars 13 forks source link

Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) #2

Closed uahsan3 closed 6 years ago

uahsan3 commented 6 years ago

Hello,

When I run the code, it throws this error:

Creating Layer conv1
I1011 13:25:12.462865  7866 net.cpp:406] conv1 <- im_concat
I1011 13:25:12.462875  7866 net.cpp:380] conv1 -> conv1
F1011 13:25:12.899116  7866 cudnn.hpp:122] Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0)  CUDNN_STATUS_BAD_PARAM
*** Check failure stack trace: ***
    @     0x7f01584a5daa  (unknown)
    @     0x7f01584a5ce4  (unknown)
    @     0x7f01584a56e6  (unknown)
    @     0x7f01584a8687  (unknown)
    @     0x7f0158b6c8f3  caffe::CuDNNConvolutionLayer<>::Reshape()
    @     0x7f0158af40fa  caffe::Net<>::Init()
    @     0x7f0158af6382  caffe::Net<>::Net()
    @     0x7f0158c08020  caffe::Solver<>::InitTrainNet()
    @     0x7f0158c08583  caffe::Solver<>::Init()
    @     0x7f0158c0885f  caffe::Solver<>::Solver()
    @     0x7f0158c16b61  caffe::Creator_SGDSolver<>()
    @           0x40d187  caffe::SolverRegistry<>::CreateSolver()
    @           0x4083be  train()
    @           0x405c7c  main
    @     0x7f0156cf4f45  (unknown)
    @           0x4065ab  (unknown)
    @              (nil)  (unknown)
Aborted (core dumped)

Do you think it's a version issue? I have cudnn version 5 and cuda version 8. My system is Ubuntu 14.04.

Thanks.

uahsan3 commented 6 years ago

Solved by updating cudNN from version 5 to 6. Thanks.