2017-11-17 18 views
0

私はFaster Rcnnをテストしています。インストールは正常です。 インストール中に、私はcudnn5.1に関する問題が1つありました。私はhereの提案に従っており、インストールは正常です。チェックに失敗しました:ステータス== CUDNN_STATUS_SUCCESS(3対0)FASTER RCNNライブラリのCUDNN_STATUS_BAD_PARAM

は今、私はその後、私はこの速くrcnnのための私のインストールと間違っている何

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) 

としてエラーが発生している

./tools/demo.py 

としてデモ・コードをテストしますか?

私はcuda8.0を持っており、libcudnn5_5.1.10-1 + cuda8.0がUbuntu16.04にインストールされています。 私はQurdo K4200グラフィックカードを持っています。

答えて

0

これは私のために機能します。 libcudnn5_5.1はCUDA7.5用です。 cudnnのユーザーガイドは、GPUとドライバの要件で確認できます。だから私はCUDA8.0のcudnnv6.0に変更しました。

その後、あなたはそのために

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

の問題に直面する可能性がありますが/ libに/ fast_rcnn/config.py-速いrcnnをPY、それが動作するようになりまし次に

__C.USE_GPU_NMS = True 

to 
__C.USE_GPU_NMS = False 

を変更する必要があります。

関連する問題