安装caffe(CPU版)

最新推荐文章于 2024-05-03 15:15:30 发布

永兴呵呵哒

最新推荐文章于 2024-05-03 15:15:30 发布

阅读量4.8k

点赞数 4

分类专栏： Ubuntu下安装东西

本文链接：https://blog.csdn.net/u014106566/article/details/85179450

版权

该文章安装的是CPU版的caffe!

一、安装依赖

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install  -y libopencv-dev
sudo apt-get install -y build-essential cmake git pkg-config
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install -y liblapack-dev
sudo apt-get install -y libatlas-base-dev 
sudo apt-get install -y --no-install-recommends libboost-all-dev
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev

sudo apt-get install -y python-pip
sudo apt-get install -y python-dev
sudo apt-get install -y python-numpy python-scipy
sudo apt-get install -y python3-dev
sudo apt-get install -y python3-numpy python3-scipy

二、下载caffe

使用一下命令下载

git clone https://github.com/BVLC/caffe.git

三、开始安装

1.进入caffe的python目录
如果需要Caffe的Python接口，切换到caffe下的python目录下，输入以下命令下载python依赖库（先安装pip）：

cd caffe/python/
sudo apt-get install python-pip
for req in $(cat requirements.txt); do pip install $req; done

2.拷贝一个安装配置文件

cp Makefile.config.example Makefile.config

3.然后修改 Makefile.config 文件，在 caffe 目录下打开该文件：

sudo gedit Makefile.config

1.将
#CPU_ONLY := 1
改为
CPU_ONLY := 1
2.应用 opencv 版本，将
#OPENCV_VERSION := 3 
修改为： 
OPENCV_VERSION := 3
3.使用 python 接口,将
#WITH_PYTHON_LAYER := 1 
修改为 
WITH_PYTHON_LAYER := 1
4.修改重要的一项
将# Whatever else you find you need goes here.下面的
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 
修改为： 
INCLUDE_DIRS := $(PYTHON