深度学习(一).Caffe安装及配置Python接口(CentOS7)

来源:互联网 发布:卓智网络是国企? 编辑:程序博客网 时间:2024/04/28 09:23


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 OpenBLASmake         #这一步是编译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-imagesudo pip install protobufvim Makefile.config
打开Makefile.config后需要修改:/usr/lib64/python2.7/site-packages/numpy/core/include  

make clean     #清除上次的编译make allmake testmake runtestmake pycaffe 
make pycaffe时会有错误出现主要是一些必要的python包没有安装。
for req in $(cat caffe/python/requirements.txt);> do> sudo pip install $req;> donemake pycaffe

结果显示成功!
进入Python中,import caffe时会出现错误,我们还需要一些修改和配置才可以在Python中使用caffe。
a.添加环境变量,可以在其他地方打开Python和import caffe
vim ~./bashrcexport PYTHONPATH=/your/path/to/caffe/python:$PYTHONPATsource ~./bashrc
b.永久保存Python接口
vim /etc/profileexport PYTHONPATH=/root/caffe/python:$PYTHONPATHsource /etc/profile
再次运行时还会出错!

原因是Python内部没有安装protobuf,安装:pip install protobuf
结果运行正常!
以上就是caffe基本的搭建!由于刚刚接触,逻辑不是十分清楚,难免会有许多的不足,希望您在看后能够指出!谢谢!






阅读全文
0 0
原创粉丝点击