ubuntu16.04 caffe

最新推荐文章于 2022-07-06 17:01:00 发布

Vegetable__

最新推荐文章于 2022-07-06 17:01:00 发布

阅读量127

点赞数

本文链接：https://blog.csdn.net/Vegetable__/article/details/107450803

版权

本文记录了在Ubuntu16.04上安装Caffe的过程，包括配置NVIDIA418.56驱动、CUDA10.0、CUDNN7.5.1和OpenCV3.4.0。在Makefile.config中调整了cudnn、opencv版本及python接口，并解决了编译及运行时遇到的错误，最终成功搭建了深度学习框架。

摘要由CSDN通过智能技术生成

已经安装了NVIDIA418.56+CUDA10.0+CUDNN7.5.1+OPENCV3.4.0（版本匹配很重要）

git clone https://github.com/BVLC/caffe.git
git clone git://github.com/BVLC/caffe.git
第一个不行用第二个
cd caffe
sudo cp Makefile.config.example Makefile.config
sudo gedit Makefile.config

修改 Makefile.config 文件内容：

1.应用 cudnn

将
#USE_CUDNN := 1
修改成： 
USE_CUDNN := 1

2.应用 opencv 版本

将
#OPENCV_VERSION := 3 
修改为： 
OPENCV_VERSION := 3

3.使用 python 接口

将
#WITH_PYTHON_LAYER := 1 
修改为 
WITH_PYTHON_LAYER := 1

4.修改 python 路径

INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 
修改为： 
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 /usr/lib/x86_64-linux-gnu/hdf5/serial

修改Makefile：

sudo gedit Makefile

将：
NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)
替换为：
NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS)

将：
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m
改为：
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

开始编译：