问题:
CUDA 由6.5升级到7.5导致原有的caffe和digits不在可用。
备注,原有的caffe和digits对应6.5,由nvidia caffe v2版本copy得来,对应的cudnn也是v2。
现在:
1)copy fast-rcnn的github文件。
git clone --recursive https://github.com/rbgirshick/fast-rcnn.git
2)安装和验证python组件包
3)Cython模块
cd $FRCN_ROOT /lib
make
4)生成caffe和pycaffe
cd $FRCN_ROOT/caffe-fast-rcnn
cat > make.log
make -j8 2>&1|tee make.log
make pycaffe 2>&1|tee make.log
对应的Makefile.config如下:
de > ## Refer to http://caffe.berkeleyvision.org/installation.html de> de > # Contributions simplifying and improving our build system are welcome! de> de > de> de > # cuDNN acceleration switch (uncomment to build with cuDNN). de> de > USE_CUDNN := 1 de> de > de> de > # CPU-on ly switch (uncomment to build without GPU support). de> de > # CPU_ON LY := 1 de> de > de> de > # To customize your choice of compiler, uncomment and set the following. de> de > # N.B. the default for Linux is g++ and the default for OSX is clang++ de> de > CUSTOM_CXX := g ++ de> de > de> de > # CUDA directory contains bin/ and lib/ directories that we need. de> de > CUDA_DIR := / usr / local / cuda de> de > # On Ubuntu 14.04, if cuda tools are installed via de> de > # "sudo apt-get install nvidia-cuda-toolkit" then use this instead: de> de > # CUDA_DIR := /usr de> de > de> de > # CUDA architecture setting: going with all of them. de> de > # For CUDA < 6.0, comment the *_50 lines for compatibility. de> de > CUDA_ARCH := - gencode arch = compute_20 , co de = sm_20 \ de> de > - gencode arch = compute_20 , co de = sm_21 \ de> de > - gencode arch = compute_30 , co de = sm_30 \ de> de > - gencode arch = compute_35 , co de = sm_35 \ de> de > - gencode arch = compute_50 , co de = sm_50 \ de> de > - gencode arch = compute_50 , co de = compute_50 de> de > de> de > # BLAS choice: de> de > # atlas for ATLAS (default) de> de > # mkl for MKL de> de > # open for OpenBlas de> de > BL