python3.5参考:https://blog.csdn.net/sunshine_zkf/article/details/82927930
python2.7参考:https://blog.csdn.net/weixin_43254559/article/details/104073838
1. 查看电脑是否有Nvidia的GPU
lspci | grep -i nvidia
结果1:如果未显示任何内容,则证明你的电脑不支持GPU。
结果2:如果显示如下图内容:
因为自己笔记本电脑没有NVIDIA的GPU,这篇文章只介绍CPU的安装。
2. 卸载Caffe的方法
直接将caffe文件夹完全删掉就行。
3. 重新安装caffe
切换root,毕竟root可以执行一切
sudo su root
---------正式开始了下面----------
3.1 安装依赖包
依次执行以下语句:
sudo apt-get install libprotobuf-dev
sudo apt-get install libleveldb-dev
sudo apt-get install libsnappy-dev
sudo apt-get install libopencv-dev
sudo apt-get install libhdf5-serial-dev
sudo apt-get install protobuf-compiler
sudo apt-get install libgflags-dev
sudo apt-get install libgoogle-glog-dev
sudo apt-get install liblmdb-dev
sudo apt-get install libatlas-base-dev
3.2 下载caffe
直接使用Git下载caffe,如果没有Git,请先安装Git,作者新装的Ubuntu16.04没有Git,所以先安装Git,执行如下语句:
sudo apt-get install git
安装结束后下载caffe,执行以下命令:
git clone https://gitee.com/chenhongqiong/caffe.git
下载后可以根目录下看到caffe文件夹
3.3 编译caffe
1.进入caffe目录
cd caffe/
2.生成Makefile.config文件,将caffe目录下自带的Makefile.config.example文件复制一份并更名为Makefile.config,命令如下:
cp Makefile.config.example Makefile.config
3.修改Makefile.config文件中的配置
1)编辑Makefile.config文件
sudo gedit Makefile.config
2)去掉CPU_ONLY前面的#号
3)配置引用文件路径(主要是HDF5的路径问题)
将下面配置改为
将上面的配置改为下面的配置
4)其他配置修改
去掉WITH_PYTHON_LAYER := 1前面的#号
5)执行编译
-
sudo make all -j8
-
sudo make test
-
sudo make runtest
执行完后显示如下结果:
有的显示1162也表示成功了。注意:如果执行结果失败,则需要执行语句sudo make clean
,然后解决掉问题重新编译
3.4 编译python接口
通常为了更好的调用caffe,我们往往需要写代码,caffe具有Python和C++接口,我们较为常用Python接口,这里我将展示如何编译Python接口。
1)安装pip3
因为Ubuntu16.04默认安装的是python2.7,没有pip3,所以需要先安装pip3,命令如下:
sudo apt-get install python3-pip
3)安装Python接口依赖库
在caffe根目录下,有个python文件夹,文件夹里面有个requirements.txt,里面有需要的依赖库和版本信息,按照其安装即可,在安装前,执行命令: gedit requirements.txt
将 python-dateutil>=1.4,<2 修改为 python-dateutil ,也就是去掉要求下载的版本号。
ctrl + S保存并关闭文件。然后需要先安装fortran编辑器(gfrotran),因为安装scipy库时需要它,命令如下:
sudo apt-get install gfortran
cd ~/caffe/python
for req in $(cat requirements.txt); do pip3 install $req; done
安装结束后,可以执行如下语句验证:
sudo pip3 install -r requirements.tx
可以看到,安装成功的,都会显示Requirement already satisfied,没安装成功的,会继续进行安装
3)将caffe根目录下的python文件夹加入到环境变量
先打开配置文件bashrc
sudo gedit ~/.bashrc
在文件的最后面添加
export PYTHONPATH=/home/电脑登录名/caffe/python:$PYTHONPATH
然后执行更新配置(有时更新不起作用时,可以重启电脑就会有效):
source ~/.bashrc
4)编译python接口
-
cd ~/caffe/
-
make pycaffe
-
如果出现“make: Nothing to be done for `pycaffe'.
-
sudo make clean之后运行
$ sudo make pycaffe
然后在命令行输入python;再输入import caffe就可以成功
5)验证python接口
进行python环境,引入caffe包,如果没有报错则安装成功!
如下图: