检查失败：status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM for FASTER RCNN Library

Question

我正在测试Faster Rcnn。安装没问题。安装过程中cudnn5.1有一个问题，我按照建议here安装好了。

现在我测试演示代码为

./tools/demo.py

然后我有错误

I1117 09:48:41.011925 12503 net.cpp:51] Initializing net from parameters: 
name: "VGG_ILSVRC_16_layers"
state {
  phase: TEST
  level: 0
}
.
.
.
layer {
  name: "cls_prob"
  type: "Softmax"
  bottom: "cls_score"
  top: "cls_prob"
}
I1117 09:48:41.012234 12503 layer_factory.hpp:77] Creating layer input
I1117 09:48:41.012251 12503 net.cpp:84] Creating Layer input
I1117 09:48:41.012259 12503 net.cpp:380] input -> data
I1117 09:48:41.012271 12503 net.cpp:380] input -> im_info
I1117 09:48:41.328574 12503 net.cpp:122] Setting up input
I1117 09:48:41.328608 12503 net.cpp:129] Top shape: 1 3 224 224 (150528)
I1117 09:48:41.328614 12503 net.cpp:129] Top shape: 1 3 (3)
I1117 09:48:41.328618 12503 net.cpp:137] Memory required for data: 602124
I1117 09:48:41.328624 12503 layer_factory.hpp:77] Creating layer conv1_1
I1117 09:48:41.328655 12503 net.cpp:84] Creating Layer conv1_1
I1117 09:48:41.328660 12503 net.cpp:406] conv1_1 <- data
I1117 09:48:41.328670 12503 net.cpp:380] conv1_1 -> conv1_1
F1117 09:48:41.676553 12503 cudnn.hpp:128] Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0)  CUDNN_STATUS_BAD_PARAM
*** Check failure stack trace: ***
Aborted (core dumped)

我安装这个更快的 rcnn 有什么问题？

我有cuda8.0，libcudnn5_5.1.10-1+cuda8.0安装在Ubuntu16.04上。我有Qurdo K4200显卡。

Answer 1

现在对我有用了。因为 libcudnn5_5.1 适用于 CUDA7.5。可以在 GPU 和驱动程序要求中查看 cudnn 的用户指南。所以我为了CUDA8.0改成了cudnnv6.0.

那么你可能会遇到

的问题

Check failed: error == cudaSuccess (8 vs. 0)  invalid device function

为此，您需要 py-faster-rcnn/lib/fast_rcnn/config.py 并更改

__C.USE_GPU_NMS = True

to 
__C.USE_GPU_NMS = False

那么现在可以了。

检查失败：status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM for FASTER RCNN Library

Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM for FASTER RCNN Library

deep-learning

caffe

pycaffe