菜鸟跑通caffe(四)——fast-cnn编译记录

问题:
CUDA 由6.5升级到7.5导致原有的caffe和digits不在可用。

备注,原有的caffe和digits对应6.5,由nvidia caffe v2版本copy得来,对应的cudnn也是v2。
现在:
1)copy fast-rcnn的github文件。
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如下: 
     
     
     
  1. de > ## Refer to http://caffe.berkeleyvision.org/installation.html de>
  2. de > # Contributions simplifying and improving our build system are welcome! de>
  3. de > de>
  4. de > # cuDNN acceleration switch (uncomment to build with cuDNN). de>
  5. de > USE_CUDNN := 1 de>
  6. de > de>
  7. de > # CPU-only switch (uncomment to build without GPU support). de>
  8. de > # CPU_ONLY := 1 de>
  9. de > de>
  10. de > # To customize your choice of compiler, uncomment and set the following. de>
  11. de > # N.B. the default for Linux is g++ and the default for OSX is clang++ de>
  12. de > CUSTOM_CXX := g ++ de>
  13. de > de>
  14. de > # CUDA directory contains bin/ and lib/ directories that we need. de>
  15. de > CUDA_DIR := / usr / local / cuda de>
  16. de > # On Ubuntu 14.04, if cuda tools are installed via de>
  17. de > # "sudo apt-get install nvidia-cuda-toolkit" then use this instead: de>
  18. de > # CUDA_DIR := /usr de>
  19. de > de>
  20. de > # CUDA architecture setting: going with all of them. de>
  21. de > # For CUDA < 6.0, comment the *_50 lines for compatibility. de>
  22. de > CUDA_ARCH := - gencode arch = compute_20 , code = sm_20 \ de>
  23. de > - gencode arch = compute_20 , code = sm_21 \ de>
  24. de > - gencode arch = compute_30 , code = sm_30 \ de>
  25. de > - gencode arch = compute_35 , code = sm_35 \ de>
  26. de > - gencode arch = compute_50 , code = sm_50 \ de>
  27. de > - gencode arch = compute_50 , code = compute_50 de>
  28. de > de>
  29. de > # BLAS choice: de>
  30. de > # atlas for ATLAS (default) de>
  31. de > # mkl for MKL de>
  32. de > # open for OpenBlas de>
  33. de > BL
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值