一开始以为是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即可。
一开始以为是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即可。