1. 程式人生 > >執行caffe時報錯:Check failed: status == CUDNN_STATUS_SUCCESS

執行caffe時報錯:Check failed: status == CUDNN_STATUS_SUCCESS

 cudnn_conv_layer.cpp:53] Check failed: status == CUDNN_STATUS_SUCCESS (6 vs. 0)  CUDNN_STATUS_ARCH_MISMATCH

原因是顯示卡的compute capability不夠,只有2.1,而cuDnn需要3.0以上

怎麼樣檢視compute capability呢?

在cuda的安裝目錄的samples目錄下,有個1_Utilities\deviceQuery,編譯裡面的工程就可以看到結果了

這時可以把cuDnn的選項去掉

在各個專案點選右鍵【屬性】->【C/C++】->【所有選項】->【前處理器定義】(preprocessordefinitions)

然後在右邊點擊向下的箭頭,【編輯】


刪除裡面的USE_CUDNN,然後重新編譯

,