工控机的环境之前已经配置好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
不正之处,请批评指正