Ubuntu16.04中caffe安装(only cpu)

准备工作:
sudo apt-get update
sudo apt-get upgrade
安装依赖项:
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 libhdf5-serial-dev
安装python:
sudo apt-get install python-dev
sudo apt install python-pip


sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

下载caffe源代码:
git clone https://github.com/BVLC/caffe.git

安装caffe:
首先修改caffe中的文件Makefile.config.example:
修改文件名为Makefile.config
打开文件修该文件中的相关选项:
因为CPU MODE, 所以在CPU_ONLY := 1前面的#要去掉.
-两个路径要改成这样:(添加后面的两个hdf5的路径, 否则编译时报hdf5错误)
# Whatever else you find you need goes here.
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu/hdf5/serial
做好准备工作后开始编译
make pycaffe
make all
make test
make runtest
结果显示ALL TESTS PASSED就安装好了, 只需要再加上一个PYTHONPATH
测试Caffe的Python接口,切换到caffe/python文件目录下,记录下来当前路径,输入以下命令:
export PYTHONPATH=/path/to/caffe/python:$PYTHONPATH
进入python环境,输入:
    import caffe  
如果没有报错,证明安装成功。
上面的方法,一旦关闭终端或者打开新终端则失效,如果放到配置文件中,可以永久有效果,命令操作如下:
    #A.把环境变量路径放到 ~/.bashrc文件中  
    sudo echo export PYTHONPATH="~/caffe/python" >> ~/.bashrc  
    #B.使环境变量生效  
    source ~/.bashrc  


以上就是整个安装过程。
但是在执行make pythoncaffe时会出现报错:
找不到hdf5
找不到hdf5-hl
解决方法为:
解决 libhdf5 报错的问题

首先,去 /usr/lib/x86_64-linux-gnu,做两个软链接:

sudo ln -s libhdf5_serial.so.10.1.0 libhdf5.so
sudo ln -s libhdf5_serial_hl.so10.0.2 libhdf5_hl.so

猜你喜欢

转载自blog.csdn.net/ultimate1212/article/details/76769020
今日推荐