剛剛發現簡易方法:
conda create -n caf python=2.7
source activate caf
conda install -c defaults libprotobuf protobuf
conda install -c defaults caffe
(over)
————————————分割常規方法—————————————————
一、下載caffe
1、將caffe複製到root目錄下
2、cd /caffe
cp Makefile.config.example Makefile.config
二、搭建虛擬環境
conda create -n caf python=2.7
三、安裝依賴庫
apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler(這一大串是一個命令)
apt-get install --no-install-recommends libboost-all-dev
apt-get install libatlas-base-dev
apt-get install python-dev
apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
apt-get install python-lmdb
apt-get install python-pip
四、修改 Makefile.config
vim Makefile.config
USE_CUDNN := 1 修改爲 USE_CUDNN := 1
OPENCV_VERSION := 3 修改爲: OPENCV_VERSION := 3
WITH_PYTHON_LAYER := 1 修改爲 WITH_PYTHON_LAYER := 1
ANACONDA_HOME := /root/anaconda3/envs/caf
PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
$(ANACONDA_HOME)/include/python2.7 \
$(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib
修改爲:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
五、安裝其他包
pip install opencv-python==3.1.0
pip install -U numpy
pip install -U scikit-image
cd /root/caffe/python
pip install -r requirements.txt
六、配置環境變量
修改.bashrc文件
export PATH=/root/anaconda3/bin:$PATH
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH
export LD_LIBRARY_PATH=/root/cudnn7.3.0/lib64:$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-9.0
export HDF5_USE_FILE_LOCKING=FALSE
export PYTHONPATH=/root/caffe/python:$PYTHONPATH
七、編譯
make all
make pycaffe
make test
make runtest
測試成功:
八、在pycharm中使用
run->edit configurations->environment variables中添加:
PYTHONPATH /root/caffe/python
LD_LIBRARY_PATH /root/cudnn7.3.0/lib64;/usr/local/cuda-9.0/lib64