直接编译caffe出现的两个问题

工控机的环境之前已经配置好ubuntu14.04+CUDA7.5+cuDNN v4,再加opencv3.1。要用ResNet做分类,需要重新编译一个caffe框架。下载BVLC/caffe,接着修改Makefile.config,再make all -j4。出现两个问题:

Probolem1: unsupported gpu architecture 'compute_60'

Solution: 注释掉config文件中CUDA_ARCH里面和compute_60相关的东西,compute_61也需要注释,猜测这是支持CUDA8.0的

Problem2:build_release/lib/libcaffe.so:undefined reference to cv::imread(cv::String const&,int)

Solution: 很多教程说配置config文件时,只改了USE_CUDNN :=1和WITH_PYTHON_LAYER:=1,其实还要查看其它设置,比如这里要去掉OPENCV_VERSION :=3之前的注释,和已经安装好的版本一致,一般默认是opencv2

不正之处,请批评指正

 

posted @ 2017-05-20 11:08  习习羊  阅读(3864)  评论(2编辑  收藏  举报