ubuntu16.04 caffe 安装

参考: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编译

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

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

sudo make runtest -j12

如图所示,即安装成功。

ubuntu16.04 caffe 安装_第1张图片

 

  • cmake编译

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

mkdir cbuild

cd cbuild

cmake ..

make all -j12

make install

如图所示,即安装成功。

ubuntu16.04 caffe 安装_第2张图片

 

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

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

3、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

你可能感兴趣的:(linux)