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即可。

发布了61 篇原创文章 · 获赞 12 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/liqiming100/article/details/83305694
今日推荐