LINUX 编译 caffe

最新推荐文章于 2021-11-25 18:28:40 发布

marshallwu1

最新推荐文章于 2021-11-25 18:28:40 发布

阅读量535

点赞数 1

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/marshallwu1/article/details/84070482

本文档详细介绍了如何在Ubuntu16.04环境下，结合CUDA9.0和CUDNN7配置编译Caffe。内容包括安装依赖包，创建和修改Makefile.config文件，特别是针对CUDA版本、OpenCV版本、Python层支持以及NUMPY的设置。在编译过程中，解决找不到hdf5库和numpy头文件的问题，确保顺利编译Caffe的libcaffe.so和pycaffe。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Ubuntu16.04 + cuda9.0 + cudnn7 于服务器上配置，参考前请对照自身需要。

安装依赖包

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 libopenblas-dev liblapack-dev libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev

创建Makefile.config文件

cp Makefile.config.example Makefile.config

修改Makefile.config

1.第一步

a.如果要使用cudnn，则将
#USE_CUDNN = 1
改为
USE_CUDNN = 1
b. 如果使用的opencv是版本3的，则将
OPENCV_VERSION = 3
改为
OPENCV_VERSION = 3
c. 如果要使用python来修改layer，则将
WITH_PYTHON_LAYER = 1
改为
WITH_PYTHON_LAYER = 1

d.因为是cuda9，所以把
CUDA_ARCH = -gencode arch=compute_20,code=sm_20 \
-gencode arch=compute_20,code=sm_21 \
-gencode arch=