LINUX 编译 caffe

Ubuntu16.04 + cuda9.0 + cudnn7 于服务器上配置,参考前请对照自身需要。安装依赖包sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compilersudo apt-get install --...
摘要由CSDN通过智能技术生成

Ubuntu16.04 + cuda9.0 + cudnn7 于服务器上配置,参考前请对照自身需要。

安装依赖包

sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

创建Makefile.config文件

cp Makefile.config.example Makefile.config

修改Makefile.config

1.第一步

a.如果要使用cudnn, 则将
#USE_CUDNN = 1
改为
USE_CUDNN  = 1
b. 如果使用的opencv是版本3的, 则将
OPENCV_VERSION = 3
改为
OPENCV_VERSION = 3
c. 如果要使用python来修改layer,则将
WITH_PYTHON_LAYER = 1
改为
WITH_PYTHON_LAYER = 1

d.因为是cuda9,所以把
CUDA_ARCH = -gencode

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值