MobileNet-SSD报错Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR

初始编译环境ubuntu16.04+CUDA8.0+Opencv3.4+cudnn5.1

编译原版的caffe-ssd,下面三项命令全部通过

make all -j16
make runtest -j16
make pycaffe

但是执行MobileNet-SSD中的demo.py时,报错Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0)  CUDNN_STATUS_INTERNAL_ERROR。

猜想一:权限问题

对于我们常玩caffe的人来说,这个错误很常见,初始都以为是时cudnn的权限问题,因此加了sudo python demo.py,但是无济于事,仍然报相同的错误。我就去cuda的samples/1_Utilities/deviceQuery目录下编译了该测试文件,执行下面两条命令,如果显示出显卡的型号,驱动,显存等相关信息,初步判断CUDA安装正常。

make
./deviceQuery

这里说一下,如果是在私人的机子上配环境,建议对cuda目录执行

sudo chmod 777 -R *

赋予普通用户访问该目录下所有文件的权限,以免后面编译caffe时碰到权限不够的问题,也方便我们后面执行caffe程序时不用每次都加sudo。当然,对于大型公用服务器就别这样搞了,随便给777权限是很容易被入侵的。

猜想2:makefile.configure中配置的显卡计算容量和显卡实际性能不匹配.

此时先查询自己的先显卡计算能力,两种方法:[1]去英伟达官网查对应型号显卡的数据;[2]和上一步类似进入自己的samples/1_Utilities/deviceQuery目录下,执行

./deviceQuery

命令,然后在输出信息中查找红线圈起来的那一项,后

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值