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
2.安装一些必用包(python-devel、python-pip、gcc、gcc-c++、git)
sudo yum install python-devel python-pip gcc gcc-c++ git
3.下载安装OpenBLAS
sudo git clone https://github.com/xianyi/OpenBLAS.git
下载完成后,开始进行安装:
cd OpenBLAS make #这一步是编译 make PREFIX=/usr/ install #安装到usr目录下面
4.下载安装Caffe
git clone https://github.com/bvlc/caffe.git
下载完成后进入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
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
结果显示成功!
进入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
再次运行时还会出错!
原因是Python内部没有安装protobuf,安装:pip install protobuf
结果运行正常!
以上就是caffe基本的搭建!由于刚刚接触,逻辑不是十分清楚,难免会有许多的不足,希望您在看后能够指出!谢谢!