1 安装caffe依赖库
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install libatlas-base-dev
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
2 安装git
sudo apt-get install git
3 安装opencv
4 下载caffe
git clone https://github.com/BVLC/caffe.git
5 修改caffe的配置文件
cpu版本与gpu版本差异不大,只需要修改一个参数
1)进入caffe目录:cd caffe/
2)生成Makefile.config文件:cp Makefile.config.example Makefile.config
3)修改Makefile.config文件中的配置
a) 编辑Makefile.config文件:sudo gedit Makefile.config也可以使用ubuntu自带的文本编辑器
b) 去掉CPU_ONLY前面的#号,cpu 版本如下
CPU_ONLY := 1
如果要编译为**gpu**(由于本人的显卡是rtx3090,因为使用其他框架,使用的cuda是11.1,编译的过程中遇到caffe与版本不兼容,不得已使用caffe,caffe使用就当嵌入式设备的一个中介,偶尔调试以下caffe)
USE_CUDNN := 1
if CUDA >= 9.0
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
c)修改coanda 路径
ANACONDA_HOME := /home/anaconda3 自己的路径
PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
$(ANACONDA_HOME)/include/python3.8 \
$(ANACONDA_HOME)/lib/python3.8/site-packages/numpy/core/include
PYTHON_LIBRARIES := boost_python38 python3.8 **这个很重要,一定要去自己所在的目录去找,不然make pycaffe 和make all都会出错**
d) 配置引用文件路径(主要是HDF5的路径问题:hdf5.h、 libhdf5.so的路径以及opencv的路径,opencv是自己修改的路径,如果默认的话,貌似不需要)
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial /usr/local/opencv4/include/opencv4/ /usr/local/opencv4/include/
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial /usr/local/opencv4/lib/
注意:可以利用sudo find / -name 文件名来查找某个文件的具体位置
e) 执行编译
sudo make all -j8 具有8核
sudo make test -j8
sudo make runtest -j8
sudo make pycaffe
安装出现的问题一般都是路径不对(库的路径不对)或者某种依赖缺少
6 路径配置
sudo vim /etc/profile
添加到最后一行
export PYTHONPATH=/home/caffe/python:$PYTHONPATH
source /etc/profile
重启电脑
7 使用
命令行使用
cd /home/caffe/python 自己的路径
python
import caffe
anconda环境使用
source activate xxx(xxx代表某一个环境)
python
import caffe
会报错缺少skimage
网速够快直接pip install scikit-image
不行的化去python官网依次下载(使用迅雷速度会提高不少)
8 pycharm的使用
file->setting->showall->漏斗旁边的,类似一棵树符号->+ 添加caffe的路径 /home/caffe/python