工控機的環境之前已經配置好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
不正之處,請批評指正
