Caffe安装及配置Python接口(CentOS7)

CentOS7下安装Caffe

1.安装Caffe依赖:

sudo yum install epel-release 为一会的其他的依赖包安装做准备
sudo yum install protobuf-devel leveldb-devel snappy-devel opencv-devel boost-devel hdf5-devel 
sudo yum install gflags-devel glog-devel lmdb-devel

Caffe安装及配置Python接口(CentOS7)

2.安装一些必用包(python-devel、python-pip、gcc、gcc-c++、git)

sudo yum install python-devel python-pip gcc gcc-c++ git

Caffe安装及配置Python接口(CentOS7)

3.下载安装OpenBLAS

sudo git clone https://github.com/xianyi/OpenBLAS.git

Caffe安装及配置Python接口(CentOS7)

下载完成后,开始进行安装:

cd OpenBLAS
make #这一步是编译
make PREFIX=/usr/ install #安装到usr目录下面

4.下载安装Caffe

git clone https://github.com/bvlc/caffe.git

Caffe安装及配置Python接口(CentOS7)

下载完成后进入Caffe目录下

cd caffe/

并且将Makefile.config.example给Makefile.config

mv Makefile.config.example Makefile.comfig

下一步需要修改Makefile.config,对Caffe进行配置。

vim Makefile.config

修改改BLAS=open,BLAS_INCLUDE=/usr/include,BLAS_LIB=/usr/lib,这是因为我们安装OpenBLAS时将其安装在/usr/下

找到#CPU_ONLY:=1将其前面的#删除,保存退出;如果需要用到GPU需要安装CUDA,安装CUDA需要安装NVIDIA驱动(仅支持NVIDIA显卡),具体方法请自行谷歌或者百毒

make all
make test

结果运行正常!

对一些文件进行配置以便Caffe的正常运行,返回到根目录下:

vim ~/.bashrc

增加下面三行

exportLD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/glog/lib

exportLD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib

exportLD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/lib

source ~/.bashrc

返回caffe,执行make runtest 结果没有问题,基础的Caffe平台已经安装好了!

5.python的接口和调用的配置caffe需要编译外壳:

在使用Python调用caffe事需要编译外壳,因此我们需要一些安装和配置,先进行一些必要的安装和配置:

sudo pip install requests[security]
sudo pip install scikit-image
sudo pip install protobuf
vim Makefile.config

打开Makefile.config后需要修改:/usr/lib64/python2.7/site-packages/numpy/core/include

Caffe安装及配置Python接口(CentOS7)

make clean #清除上次的编译
make all
make test
make runtest
make pycaffe

make pycaffe时会有错误出现主要是一些必要的python包没有安装。

for req in $(cat caffe/python/requirements.txt);
> do
> sudo pip install $req;
> done
make pycaffe

Caffe安装及配置Python接口(CentOS7)

Caffe安装及配置Python接口(CentOS7)

结果显示成功!

进入Python中,import caffe时会出现错误,我们还需要一些修改和配置才可以在Python中使用caffe。

a.添加环境变量,可以在其他地方打开Python和import caffe

vim ~./bashrc
export PYTHONPATH=/your/path/to/caffe/python:$PYTHONPAT
source ~./bashrc

b.永久保存Python接口

vim /etc/profile
export PYTHONPATH=/root/caffe/python:$PYTHONPATH
source /etc/profile

再次运行时还会出错!

Caffe安装及配置Python接口(CentOS7)

原因是Python内部没有安装protobuf,安装:pip install protobuf

Caffe安装及配置Python接口(CentOS7)

结果运行正常!

以上就是caffe基本的搭建!由于刚刚接触,逻辑不是十分清楚,难免会有许多的不足,希望您在看后能够指出!谢谢!

相关推荐