查看: 1833|回复: 1

安装Caffe

[复制链接]

665

主题

1234

帖子

6575

积分

xdtech

Rank: 5Rank: 5

积分
6575
发表于 2020-3-13 21:06:04 | 显示全部楼层 |阅读模式
7.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
sudo apt-get install python-pip
sudo apt-get install gfortran
sudo apt-get install python-protobuf
sudo apt-get install python-skimage
sudo apt-get install --no-install-recommends libboost-all-dev
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
7.2.使用Git 从github上下载Caffe ,
地址:https://github.com/BVLC/caffe
在终端执行命令:

git clone https://github.com/BVLC/caffe
1
下载完成后,在home文件下面用Caffe文件夹

我们接下来按照下面的步骤进行安装Caffe:
1.生成Makefile.config文件,将caffe目录下自带的Makefile.config.example文件复制一份并更名为Makefile.config,命令如下:

cp Makefile.config.example Makefile.config
1
2.修改配置文件
执行下面命令,打开makefile.config文件:

sudo gedit Makefile.config
1
根据自己情况修改:
a.若使用cudnn,将:
#USE_CUDNN := 1
修改为:
USE_CUDNN := 1
b.使用opencv3版本 则:
将:
#OPENCV_VERSION := 3
修改为:
OPENCV_VERSION := 3
c.使用Python接口:
#WITH_PYTHON_LAYER := 1
修改为
WITH_PYTHON_LAYER := 1
d.修改Python路径,将:

NCLUDE_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  
1
2
3
4
5
e.去掉computer_20:
找到:

CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
            -gencode arch=compute_20,code=sm_21 \
            -gencode arch=compute_30,code=sm_30 \
            -gencode arch=compute_35,code=sm_35 \
            -gencode arch=compute_50,code=sm_50 \
            -gencode arch=compute_52,code=sm_52 \
            -gencode arch=compute_60,code=sm_60 \
            -gencode arch=compute_61,code=sm_61 \
            -gencode arch=compute_61,code=compute_61
1
2
3
4
5
6
7
8
9
修改为:

CUDA_ARCH := -gencode arch=compute_30,code=sm_30 \
            -gencode arch=compute_35,code=sm_35 \
            -gencode arch=compute_50,code=sm_50 \
            -gencode arch=compute_52,code=sm_52 \
            -gencode arch=compute_60,code=sm_60 \
            -gencode arch=compute_61,code=sm_61 \
            -gencode arch=compute_61,code=compute_61
1
2
3
4
5
6
7
因为从CUDA9.0后就不在支持compute_20了,这里不修改的话编译的时候就会报错;
3.到这里我们就可以编译Caffe了,依次执行下面命令:

make  all -j8
make pycaffe
make test -j8
make runtest -j8
1
2
3
4
如果上面的步骤都按照完成的话,make就不会出错,只要make不出错的话,下面的三句命令就会顺利编译完成,需要一点时间。如果make错误的话,针对错误的命令去找解决,一定要耐心,都是可以解决的。
4.修改环境变量:
就是将caffe中的Python导入到解释器中。
执行下面命令打开配置文件bashrc

sudo gedit ~/.bashrc
1
在文件最后添加:

export PYTHONPATH=/home/zixiang/caffe/pythonPYTHONPATH
1
在这里一定要注意不能直接复制,这里的路径是自己home文件夹安装Caffe时里面Python接口的路径。所以根据自己安装的地址修改即可,一般只要对应的修改我这个路径里面的zixiang这个名称位置即可。
修改后保存退出,执行下面预命令让修改生效

sudo source ~/.bashrc
1
5.验证Python接口
在终端中打开Python环境,在Python环境下输入下面命令:

import caffe
1
没有报错的话就说明Caffe已经安装成功,到这里我们就已经配置好全部的环境了,下面就可以针对自己的目的需要去下载其他的算法。配置环境道理不容易,要耐心,心细。

Caffe中遇到的错误及解决方案:

1.[Errno 13] 权限不够: '/usr/local/lib/python2.7/dist-packages/cloudpickle’
ERROR::未找到命令
解决方案:在安装软件的时候在install后面加上 --user即可

pip install --user scikit-image
1
2.Traceback (most recent call last): File “/usr/bin/pip”, line 9, in < module >
解决方案 :
将 /usr/bin/pip 文件中:

from pip import  main
if __name__ == '__main__':
    sys.exit(main())
1
2
3
改为:

from pip import __main__
if __name__ == '__main__':
    sys.exit(__main__._main())
1
2
3
3.ImportError: No module named skimage.io
错误提示:

>>> import caffe
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "caffe/__init__.py", line 1, in <module>
    from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
  File "caffe/pycaffe.py", line 15, in <module>
    import caffe.io
  File "caffe/io.py", line 2, in <module>
    import skimage.io
ImportError: No module named skimage.io
1
2
3
4
5
6
7
8
9
10
解决方案;

缺乏skimage工具箱

pip install  --user scikit-image
1
参考博客:
https://blog.csdn.net/A_Z666666/article/details/72853346
https://blog.csdn.net/la_fe_/article/details/81516370
https://blog.csdn.net/balixiaxuetian/article/details/79154013


回复

使用道具 举报

665

主题

1234

帖子

6575

积分

xdtech

Rank: 5Rank: 5

积分
6575
 楼主| 发表于 2020-3-13 21:06:14 | 显示全部楼层
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表