Caffe Python開發環境設定__Python

安裝 Python 依賴的庫

由於安裝 Python 的第三方庫的時候需要編譯，所以需要安裝下面兩個庫

$ sudo yum install python-devel numpy

設定virtualenv環境

$ virtualenv caffeenv$ cd caffeenv$ . bin/activate

安裝 Python 第三方庫

cd <path_to_caffe>/caffe-1.0/pythonpip install -r requirements.txt

編譯 pycaffe

$ make pycaffe

測試

首先需要設定環境變數 PYTHONPATH，如下：

$ export PYTHONPATH=<path_to_caffe>/caffe-1.0/python:$PYTHONPATH

運行python，進入互動模式，然後使用 “import caffe” 來測試環境。

import caffe

問題

在搭建環境的時候也碰到了一些問題，現總結如下： 問題一

在 “import caffe” 的時候出現下面錯誤

ImportError: No module named _caffe

問題原因是沒有將caffe加入到 PYTHONPATH 環境變數裡。

解決辦法是在運行 python 之前，設定 PYTHONPATH 環境變數。

export PYTHONPATH=/home/jhadmin/Downloads/caffe-1.0/python:$PYTHONPATH

問題二

在 “import caffe” 的時候出現下面錯誤

AttributeError: 'module' object has no attribute 'bool_'

解決辦法

$ make pycaffe

問題三

在 “import caffe” 的時候出現下面錯誤

CXX/LD -o python/caffe/_caffe.so python/caffe/_caffe.cpppython/caffe/_caffe.cpp:10:31: fatal error: numpy/arrayobject.h: No such file or directory #include <numpy/arrayobject.h>

解決辦法

sudo yum install numpy