Ubuntu下安装caffe

来源:互联网 发布:两张excel找出相同数据 编辑:程序博客网 时间:2024/04/29 18:48
1.先安装依赖项目和下载源代码
写一个脚本,然后bash 这个脚本执行下。

#install caffe in ubuntu
#!/bin/sh
sudo apt-get install git
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 libatlas-base-dev
sudo apt-get install python-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
git clone https://github.com/bvlc/caffe.git


2.修改配置文件
打开caffe文件夹
#cd caffe
并修改配置文件
把Makefile.config.example命名为Makefile.config
#mv Makefile.config.example Makefile.config

修改Makefile.config文件:
找到
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
修改为
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/include/hdf5/serial
如果没有支持cuda的卡,取消掉 CUP_ONLY 前面的符号 #

(如果权限不够,请先执行命令
#chmod -R 777 Makefile.config




修改Makefile文件:
找到
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
后面两项变为
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

3.编译
不知道是不是因为电脑性能不够,我是在切换tty下进行的。具体方法是按下 ctrl+alt+f1 登录ubuntu后 cd caffe -> make all
0 0
原创粉丝点击