使用caffe训练faster-rcnn时遇到的问题总结

经过今天一天的实验,充分了解了在深度学习领域,显存大的重要性。 不得不承认,有了N卡以后,效率提高了很多,当然这也得益于纠缠于cpu训练faster rcnn的日子.... (1)今天早上,主要解决了faster rcnn 下的caffe不能编译的问题。 问题产生的原因是faster rcnn下的caffe不支持cudnn5.1,当时就单纯的把Makefile.config下的 USE_CUDNN
相关文章
相关标签/搜索