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基本的搭建!由於剛剛接觸,邏輯不是十分清楚,難免會有許多的不足,希望您在看後能夠指出!謝謝!
TAG:程序員小新人學習 |