CentOS 6.6 安装Caffe
条评论为了能够测试和使用caffe,需要将caffe环境配置到服务器上,之前在自己的ubuntu上安装过caffe,但中间有一些步骤没法通过,最终虽然勉强安装上了,但总感觉有点问题,这次在服务器上安装,又找了一些资料加上上次的经验,最终成功安装上了,特此记录一下,方便其他人安装调试。
服务器上安装有Intel的parallel studio 2015,因此在安装的时候可以不用为mkl担心,不过安装这个也不麻烦,下载一个包自己安装就是了。
主要参考了官方网站的指南.
安装依赖
1 | sudo yum install protobuf-devel leveldb-devel snappy-devel opencv-devel boost-devel hdf5-devel |
我在安装的时候,提醒glog没找到,因此手动安装glog
1 | # glog |
由于caffe会依赖opencv,所以还需要安装opencv的库,先安装opencv的依赖:
1 | sudo yum install cmake git pkgconfig gtk2 gtk2-devel |
opencv的安装
Caffe作者默认你已经配置好了OpenCV环境,文档里没有说这一步。好在有人已经写好了配置OpenCV的脚本 ,直接拿来用。或者可以按照官方教程来进行安装
1 | git clone https://github.com/jayrambhia/Install-OpenCV |
安装的时候,由于我装的cuda 7.0已经不支持计算能力1.X的版本了,为此opencv中编译时可能会报各种各样的错误,解决办法就是cmake下OpenCVDetectCUDA.cmake 文件,找到set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} ${NVCC_FLAGS_EXTRA}),手动改成:
1 | set(CUDA_NVCC_FLAGS -gencode;arch=compute_20,code=sm_20;-gencode;arch=compute_20,code=sm_21;-gencode;arch=compute_30,code=sm_30;-gencode;arch=compute_35,code=sm_35;-gencode;arch=compute_30,code=compute_30) |
至少这样可以编译过了。编译完了make install就可以了。
cudnn的安装
到cuda官网下载cudnn。
解压后将cudnn.h复制到/usr/local/cuda/include/下,将libcudnn.so libcudnn_static.a 复制到/usr/local/cuda/lib64/下
1 | ln -s libcudnn.so.6.5.48 libcudnn.so.6.5 |
protobuf的安装
protobuf是google公司的一个开源项目,主要功能是把某种数据结构的信息以某种格式保存及传递,类似微软的XML,但效率较高。目前提供C++,java和python的API。caffe需要在python中调用protobuf,因此我们需要安装protobuf的python版
1 | wget https://github.com/google/protobuf/releases/download/v3.0.0-alpha-2/protobuf-python-3.0.0-alpha-2.tar.gz |
安装速度比较慢,我下了好久才下下来。完成安装后,还需要去python文件夹内安装python所需要的模块
1 | cd python |
没有报错说明安装正常。
caffe安装
在caffe文件夹里,先复制一份Makefile.config cp Makefile.config.example Makefile.config 修改Makefile.config
1 cudnn 注释掉USE_CUDNN:=1
2 使用intel 的mkl 来计算blas 修改为: BLAS:=mkl
接下来就开始进行编译并测试
1 | cp Makefile.config.example Makefile.config |
但在进行编译时,总是提示boost_thread有问题,因此重新安装boost库,先从官网下载下来
1 | tar zxvf boost_1_57_0.tar.gz |
看到成功提示:
The Boost C++ Libraries were successfully built!
在caffe目录下make clean后重新编译,即可。
本文标题:CentOS 6.6 安装Caffe
文章作者:throneclay
发布时间:2015-10-09
最后更新:2022-08-03
原始链接:http://blog.throneclay.top/2015/10/09/cic/
版权声明:本博客所有文章除特别声明外,均采用 CC BY-NC-SA 3.0 CN 许可协议。转载请注明出处!