當前位置:
首頁 > 知識 > Caffe安裝及配置Python介面(CentOS7)

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

喜歡這篇文章嗎?立刻分享出去讓更多人知道吧!

本站內容充實豐富,博大精深,小編精選每日熱門資訊,隨時更新,點擊「搶先收到最新資訊」瀏覽吧!


請您繼續閱讀更多來自 程序員小新人學習 的精彩文章:

EF調用存儲過程實現分頁
linux虛擬地址轉物理地址

TAG:程序員小新人學習 |