0.前言
- 从ssd作者的github上clone的代码跑MobileNet-SSD会出现错误:
Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR
1.原因
- dw卷积gpu模式有bug,不使用cudnn就可以运行,但速度感人
2.解决方法
git clone https://gitee.com/awake658/caffe
cd caffe
git checkout ssd
编译,和caffe一样
cd examples/MobileNet-SSD
python2 demo.py
3. 20210429
- 前几天训练MobileNet-YOLO时,使用这两个文件之后还是有问题,于是
- 使用cuda不使用cudnn重新编译caffe
- 成功运行,速度也挺快