我正在测试Faster Rcnn。安装没问题。在安装过程中,我遇到了 cudnn5.1 的一个问题,我遵循了这里的建议,现在安装很好。
现在我将演示代码测试为
./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 显卡。
现在它对我有用。由于 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
然后现在它起作用了。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句