ubuntu16.04 caffe 安装

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

CYummy

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

阅读量1k

点赞数

分类专栏： linux 文章标签： linux caffe

本文链接：https://blog.csdn.net/CYummy/article/details/82789355

版权

linux 专栏收录该内容

7 篇文章 0 订阅

订阅专栏

参考：https://blog.csdn.net/yhaolpz/article/details/71375762　　该博主写的很详细

我的配置：ubuntu 16.04 + cuda9.0 + cudnn7.1 + opencv3.3.0

tips: caffe可以卸掉重新安装，只要把原caffe文件夹删掉再按步骤重新走一遍就可以了。

caffe可以选用make和cmake两种方式进行编译安装。make和cmake编译互不影响，之所以需要cmake编译是因为楼主用Cmakelist.txt管理工程，需要添加caffe的头文件和相应的库，使用cmake编译非常方便。

参考：https://blog.csdn.net/u013832707/article/details/53159071

make编译

将Ｍakefile和Makefile.config两个文件修改好后，直接在同级目录下调用终端，输入命令：

make all -j12 根据自己电脑cpu核数修改-j12

sudo make runtest -j12

如图所示，即安装成功。

cmake编译

创建cbuild文件夹，区别与make编译的build文件夹。输入命令：

mkdir cbuild

cd cbuild

cmake ..

make all -j12

make install

如图所示，即安装成功。

在自己实际安装过程中，遇到了些问题，在此记录下：

1、根据自己cuda的安装版本，将Makefile.config文件做相应的更改，即按照如下要求将不需要的部分注释掉。

# CUDA architecture setting: going with all of them.
# For CUDA < 6.0, comment the *_50 through *_61 lines for compatibility.
# For CUDA < 8.0, comment the *_60 and *_61 lines for compatibility.
# For CUDA >= 9.0, comment the *_20 and *_21 lines for compatibility.

2、Check failed: status == CURAND_STATUS_SUCCESS (201 vs. 0) CURAND_STATUS_LAUNCH_FAILURE

该问题参考github得到解决，大致原因是cuda旧版本库没有清理干净，找到相应文件夹（/usr/lib/x86_64-linux-gnu），将旧版本的.so文件删除（比如： *.so.7.5），再次编译，成功安装。

链接：https://github.com/BVLC/caffe/issues/5195

３、caffe/proto/caffe.pb.h: 没有那个文件或目录

# In the directory you installed Caffe to
protoc src/caffe/proto/caffe.proto --cpp_out=.
mkdir include/caffe/proto
mv src/caffe/proto/caffe.pb.h include/caffe/proto

参考：https://github.com/muupan/dqn-in-the-caffe/issues/3