1. 程式人生 > >Check failed: status == CUDNN_STATUS_SUCCESS (6 vs. 0)

Check failed: status == CUDNN_STATUS_SUCCESS (6 vs. 0)

一開始以為是batch size設定得太大,將batch size 設定為 1 後出現如下得錯誤。

CUDNN_STATUS_SUCCESS (1 vs. 0) CUDNN_STATUS_NOT_INITIALIZED

修改CMakeLists.txt的第41行,即如下的行;

caffe_option(USE_CUDNN "Build Caffe with cuDNN library support" OFF IF NOT CPU_ONLY)

把ON改為OFF即可。