cudnn_conv_layer.cpp:53] Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR

dog-qiuqiu / MobileNet-Yolo

MobileNetV2-YoloV3-Nano: 0.5BFlops 3MB HUAWEI P40: 6ms/img, YoloFace-500k:0.1Bflops 420KB:fire::fire::fire:

Other

1.71k stars 277 forks source link

cudnn_conv_layer.cpp:53] Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0) CUDNN_STATUS_INTERNAL_ERROR #5

Open liuwei3612 opened 4 years ago

liuwei3612 commented 4 years ago

python darknet2caffe.py MobileNetV2-YOLOv3-Nano-voc.cfg MobileNetV2-YOLOv3-Nano-voc.weights MobileNetV2-YOLOv3-Nano-voc.prototxt MobileNetV2-YOLOv3-Nano-voc.caffemodel

dog-qiuqiu commented 4 years ago

hi，你好，你编译caffe-upsample要编译cpu版本的，否则模型转换的时候会爆显存，这是因为caffe对于分组卷积支持不好。

liuwei3612 commented 4 years ago

不好意思，这里我有点不明白，是在编译caffe的时候就只编译cpu版本的吗？

dog-qiuqiu commented 4 years ago

是的，编译cpu版本的caffe

liuwei3612 commented 4 years ago

好的，我来试试，谢谢！