一直想用caffe,一直没有静下心好好的研究,今天下午看看了caffe。现在粉上了贾扬清大神。贴上caffe官网。
caffe是贾大神在博士论文阶段写的一个框架,对大神真是膜拜。
这个框架有两种安装方式,一种是make,这种是官方提供的方法,一种是cmake方式,这种是caffe开源社区贡献者提供的。现在记录一下两种方式安装方法,首先贴上官方安装教程。
首先下载caffe:
1 2 |
git clone https://github.com/BVLC/caffe.git |
Compilation with Make
- 复制Makefile.config
12sudo cp Makefile.config.example Makefile.config - 修改Makefile.config
- 打开cudnn
12345将#USE_CUDNN := 1修改成:USE_CUDNN := 1 - 指定opencv版本
12345将#OPENCV_VERSION := 3修改为:OPENCV_VERSION := 3 - 使用python接口
12345将#WITH_PYTHON_LAYER := 1修改为WITH_PYTHON_LAYER := 1 - 修改python路径
123456INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/includeLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib修改为:INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serialLIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
- 打开cudnn
- 修改Makefile
- 改NVCCFLAGS
12345将:NVCCFLAGS +=-ccbin=$(CXX) -Xcompiler-fPIC $(COMMON_FLAGS)替换为:NVCCFLAGS += -D_FORCE_INLINES -ccbin=$(CXX) -Xcompiler -fPIC $(COMMON_FLAGS) - 改LIBRARIES
12345将:LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5改为:LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial
- 改NVCCFLAGS
- make
1234make allmake testmake runtest
CMake Build
cmake后一定要看cmake的输出,然后有什么错就解决什么错,因为这种即使cmake成功,后面make也会报错!
cmake安装不用改什么Makefile.config和Makefile,但是要确保环境安装好,我由于是之前安装opencv cmake的时候把cuda选项关掉了,所以一直报错。所以针对cmake报错什么解决什么。然后正常cmake安装就行。
1 2 3 4 5 6 7 |
mkdir build cd build cmake .. make all make install make runtest |
最后出现下面的就代表安装成功。
最新评论
感谢博主,让我PyTorch入了门!
博主你好,今晚我们下馆子不?
博主,你的博客用的哪家的服务器。
您好,请问您对QNN-MO-PYNQ这个项目有研究吗?想请问如何去训练自己的数据集从而实现新的目标检测呢?
where is the source code ? bomb1 188 2 8 0 0 hello world 0 0 0 0 0 0 1 1 9?5
在安装qemu的过程中,一定在make install 前加入 sudo赋予权限。
所以作者你是训练的tiny-yolov3还是yolov3...
很有用