运行caffe时报错:Check failed: status == CUDNN_STATUS_SUCCESS

来源:互联网 发布:win10升级软件 编辑:程序博客网 时间:2024/06/04 01:20

 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,然后重新编译

,

0 0