ubuntu安装caffe配置makefile.config和makefile文件

最新推荐文章于 2021-10-12 10:50:38 发布

原创最新推荐文章于 2021-10-12 10:50:38 发布 · 2.7k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#Ubuntu #caffe配置

本文详细介绍了如何在Ubuntu系统中配置Caffe深度学习框架，并确保其能够与CUDA9.2、CUDNN7、GCC5.4.0和OpenCV3.4.0等软件兼容运行。文章提供了安装依赖包、配置Makefile.config和Makefile的具体步骤。

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

本文参考了
https://blog.youkuaiyun.com/autocyz/article/details/51783857
https://blog.youkuaiyun.com/shanglianlm/article/details/79292452
http://www.cnblogs.com/zjutzz/p/5716453.html

前言：
成功安装ubuntu+cuda9.2+cudnn7 +gcc 5.4.0+OpenCV 3.4.0

配置caffe

首先，是安装依赖包，参考上述链接

sudo apt-get install -y libprotobuf-dev 
sudo apt-get install -y libleveldb-dev 
sudo apt-get install -y libsnappy-dev 
sudo apt-get install -y libhdf5-serial-dev 
sudo apt-get install -y protobuf-compiler
sudo apt-get install -y libatlas-base-dev
sudo apt-get install -y  libboost-all-dev
sudo apt-get install -y libgflags-dev 
sudo apt-get install -y libgoogle-glog-dev 
sudo apt-get install -y liblmdb-dev
sudo apt-get install -y libopencv-dev

接下来就是配置Makefile.config的修改了

①将
USE_CUDNN=1
USE_OPENCV=1
的注释取消

②Caffe-GPU编译问题：nvcc fatal : Unsupported gpu architecture 'compute_20'
因为我的是cuda是9，所以要修改下面的配置，不同的cuda版本的修改不一样
将
CUDA_ARCH :=中的
-gencode arch=compute_20,code=sm_20 \
-gencode arch=compute_20,code=sm_21 \
给注释或者删除

③将
BLAS=atlas改成BLAS=open
在改完之后需要安装一个库
sudo apt-get install libopenblas-dev
或者你可以用其他的库，然后适当修改，参考上述链接，有解释该库的作用，也有整体解释Makelife.config的注释，最好可以看看

④fatal error: hdf5.h: 没有那个文件或目录
主要是因为hdf5这个库找不着，需要修改
在原来 
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial 
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib 
现在改成（64位系统）：
 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

接下来就是修改Makefile的文件了，记住不要和上面的Makefile.config混在一起

⑤make: *** [.build_release/lib/libcaffe.so.1.0.0-rc3] Error 1
将
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
改成
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

然后就完成了caffe文件的配置了！

本人新手，可能略有不足之处，请见谅！仅做参考