首先修改配置檔案,回到caffe目錄
cd ~/caffe
cp Makefile.config.example Makefile.config
gedit Makefile.config
這裡僅需修改兩處:
i) 使用cuDNN
# USE_CUDNN := 1
這裡去掉#,取消註釋為
USE_CUDNN := 1
ii) 修改python包目錄,這句話
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include
改為
/usr/local/lib/python2.7/dist-packages/numpy/core/include
因為新安裝的python包目錄在這裡: /usr/local/lib/python2.7/dist-packages/
接下來就好辦了,直接make
make all -j4
make test
make runtest
make pycaffe
這時候cd 到caffe 下的 python 目錄,試試caffe 的 python wrapper安裝好沒有:
python
import caffe
如果不報錯,那就說明安裝好了。
首先修改配置檔案,回到caffe目錄
cd ~/caffe
cp Makefile.config.example Makefile.config
gedit Makefile.config
這裡僅需修改兩處:
i) 使用cuDNN
# USE_CUDNN := 1
這裡去掉#,取消註釋為
USE_CUDNN := 1
ii) 修改python包目錄,這句話
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/lib/python2.7/dist-packages/numpy/core/include
改為
PYTHON_INCLUDE := /usr/include/python2.7 \
/usr/local/lib/python2.7/dist-packages/numpy/core/include
因為新安裝的python包目錄在這裡: /usr/local/lib/python2.7/dist-packages/
接下來就好辦了,直接make
make all -j4
make test
make runtest
make pycaffe
這時候cd 到caffe 下的 python 目錄,試試caffe 的 python wrapper安裝好沒有:
python
import caffe
如果不報錯,那就說明安裝好了。