使用caffe訓練faster-rcnn時遇到的問題總結

經過今天一天的實驗,充分了解了在深度學習領域,顯存大的重要性。 不得不承認,有了N卡以後,效率提高了很多,當然這也得益於糾纏於cpu訓練faster rcnn的日子.... (1)今天早上,主要解決了faster rcnn 下的caffe不能編譯的問題。 問題產生的原因是faster rcnn下的caffe不支持cudnn5.1,當時就單純的把Makefile.config下的 USE_CUDNN
相關文章
相關標籤/搜索