检查失败：状态 == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM 用于 FASTER RCNN 库答案

【问题标题】：Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM for FASTER RCNN Library检查失败：状态 == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM 用于 FASTER RCNN 库
【发布时间】：2018-04-30 17:39:30
【问题描述】：

我正在测试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 显卡。

【问题讨论】：

标签： deep-learning caffe pycaffe

【解决方案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

那么现在它可以工作了。

【讨论】：