初始编译环境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
命令,然后在输出信息中查找红线圈起来的那一项,后