之前在搞ssd的时候没出问题,后来重装啦一下系统,把它拷回来,发现出了点问题,在训练或者测试的时候,需要输入:
python examples/ssd/ssd_pascal.py 或者python examples/ssd/score_ssd_coco.py,但是这时会出现这个错误:
Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR
这个错误搞过caffe的都知道,没有权限问题,于是我就加上python,改成:
sudo python examples/ssd/ssd_pascal.py 或者
sudo python examples/ssd/score_ssd_coco.py,然而这时候又出现新的问题:no module named caffe,
这个感觉不应该啊,明明之前把pycaffe的路径放到bashrc里面啦。
后来sudo python ,打开输入import caffe 发现果然没有,no module named caffe
而python,import caffe就很正常。
找到这个问题的方法是:
python,import sys,sys.path,看输出的路径。
sudo python,import sys,sys.path,看输出的路径。发现这两个并不一样。python里的有caffe的路径,而sudo python没有。
解决办法:
新建一个mytest.pth,打开,把你的caffe——python路径拷进去,比如我的:/home/zwj/Documents/ssd/caffe/python,把这个文件拷到/usr/lib/python2.7/dist-packages,重新打开终端运行sudo python examples/ssd/score_ssd_coco.py不再出问题啦。大功告成!