- 前期准备说明:
由于caffe需要使用opencv库,所以需要先安装opencv,再安装caffe。
为方便各版本的深度学习框架切换,本次教程将基于anaconda下的虚拟环境进行安装。 - anconda 虚拟环境安装
假设已安装好anaconda,本文使用的python版本为3.5。
首先创建虚拟环境conda create -n your_env_name python=X.X
,比如conda create -n pytorch1 python=3.5
。
其他重要命令:
#激活虚拟环境
source activate your_env_name(修改为你的虚拟环境名称)
#退出虚拟环境
source deactivate
#删除虚拟环境
conda remove -n your_env_name(修改为你的虚拟环境名称) --all
-
opencv安装(源码编译)
难点有2个,找到适合自己电脑和需求的opencv版本,设置源码编译的路径。
参考教程:ubuntu16.04安装opencv完整教程注意将教程中的虚拟环境改为自己的conda建立的虚拟环境即可。
-
caffe安装(源码编译)
参考教程:手把手教你在Ubuntu 16.04下配置GPU版Caffe
从step3开始参考。
除教程中出现的错误及修改方法外,我安装中遇到的报错和坑记录如下(如不特别指明,修改内容均为makefile.config):
(1)fatal error: pyconfig.h: No such file or directory #include “pyconfig.h”
参考:caffe编译:fatal error: pyconfig.h: No such file or directory #include “pyconfig.h”
解决方法:将caffe/makefile.config文件的python include 修改为如下:
PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
$(ANACONDA_HOME)/include/python3.5 \
$(ANACONDA_HOME)/lib/python3.5/site-packages/numpy/core/include \
/usr/include/python3.5
(2)nvcc fatal : Unsupported gpu architecture ‘compute_20’<