Ubuntu下安装caffe

如果操作系统是ubuntu 17.04以上,安装很方便,只需要一条指令即可。
带CUDA版本的caffe:
sudo apt install caffe-cuda

只用CPU的版本:
sudo apt install caffe-cpu


ubuntu 17.04以下的版本:

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


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

猜你喜欢

转载自blog.csdn.net/sk_U238/article/details/52366067
今日推荐