caffe运行过程中遇到的问题及解决办法

问题1:用caffe处理MobileNet-SSD,且卷积层使用cudnn时,报错"cudnn_conv_layer.cpp:53] Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0)  CUDNN_STATUS_INTERNAL_ERROR"

解决办法:MobileNet-SSD的卷积层不用cudnn,而用caffe cuda版本

解释:这是由于MobileNet-SSD的卷积层中有大量的带group参数的卷积,cudnn在处理这种卷积时需要的内存较多,本人用的GTX1070Ti卡8G显存都不够用.在出现这个问题可以用nvidia-smi查看此时显存占用.

猜你喜欢

转载自blog.csdn.net/maidabu/article/details/86650453