正文

ubuntu16.04下配置caffe无GPU

IP云
IP云V管理员/1 阅读
12/02
提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

1. 安装依赖项 

1 sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler

2 sudo apt-get install --no-install-recommends libboost-all-dev

3 sudo apt-get install libopenblas-dev liblapack-dev libatlas-base-dev

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

2.  下载解压OpenCV3.1

①   OpenCV 下载链接： http://opencv.org/releases.html

下载3.1.0 Sources源码：

②   下载的OpenCV源码的压缩包 opencv-3.1.0.zip 的保存路径默认是在 home/用户名/下载 下，进入到该目录。

解压并拷贝到home目录后重命名： 

cd ~/下载

unzip opencv-3.1.0.zip

sudo mv ./opencv-3.1.0 ~/OpenCV

③  进入OpenCV目录，创建build文件夹： 

cd ~/opencv

mkdir build

cd build

3. 使用CMake编译OpenCV 

sudo apt install cmake

sudo cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..

sduo make -j4

之后编译成功，输入以下指令完成安装： 

sudo make install

4. 验证OpenCV是否正确安装

切换到OpenCV下的samples目录，并编译samples程序： 

cd ~/OpenCV/samples

sudo cmake .

sudo make -j $(nproc)

继续切换到cpp目录下，运行测试程序 

cd cpp/

./cpp-example-facedetect lena.jpg

这是一个人脸检测的测试程序，运行结果：

5.  下载caffe

caffe源代码下载地址：  https://github.com/BVLC/caffe

将下载的caffe源码压缩包解压并移动到home目录下： 

cd ~/下载

unzip caffe-master.zip

sudo mv ./caffe-master ~/caffe-master

6. 修改Makefile.config配置文件

复制一份Makefile.config文件并使用gedit打开修改配置： 

cd ~/caffe-master

sudo cp Makefile.config.example Makefile.config

sudo gedit Makefile.config

修改如下节点：

  • #USE_CUDNN:=1 改为 USE_CUDNN：=0   （不使用CUDN，注意去掉前边的#符号）
  • #OPENCV_VERSION := 3  OPENCV_VERSION := 3
  • #WITH_PYTHON_LAYER :=1 改为  WITH_PYTHON_LAYER :=1（python接口）

ubuntu16.04版本相比14.04版本，文件包含位置有变化，所以以下两处一定要做更改：

INCLUDE_DIRS := $(PYTHON_INCLUDE)
/usr/local/include 改为：

INCLUDE_DIRS := $(PYTHON_INCLUDE)
/usr/local/include /usr/include/hdf5/serial

LIBRARY_DIRS := $(PYTHON_LIB)
/usr/local/lib /usr/lib  改为：

LIBRARY_DIRS := $(PYTHON_LIB)
/usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial

7. 修改makefile文件

打开caffe-master目录下的makefile文件，做如下修改：

NVCCFLAGS +=-ccbin=$(CXX)
-Xcompiler-fPIC $(COMMON_FLAGS)  修改为：

NVCCFLAGS += -D_FORCE_INLINES
-ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

8. 编译caffe

在caffe-master目录运行终端，输入： 

sudo make all -j4

9. 测试caffe是否安装成功

运行以下指令： 

sudo make runtest

运行之后有以下结果，说明caffe配置成功：

相关阅读

目录[+]