原文 http://blog.csdn.net/u011635764/article/details/52831167
刚入坑,有什么错误请一定告诉我…
这篇博客主要是记录如何安装 python版的faster-RCNN
,因为之前已经安装过caffe了,
Ubuntu16.04系统,显卡GTX1060以各种依赖,cuda8.0, cudnn, MKL, opencv3.1.0,
都已经安装过了,python环境是Anaconda,之后有机会补上前面的各种坑吧.
另外欢迎访问我的博客,实在是百度不到的说 …. http://hammer2505.cn/
安装python依赖包
- 1
- 2
下载源代码
- 1
修改并编译Cython
- 1
修改setup.py
文件第135行
- 1
根据自己的显卡计算能力进行修改,查询网址为:CUDA GPUs
例如我的电脑是GTX 1060, 计算能力为6.1,则修改为
- 1
保存后执行
- 1
进行编译.
编译caffe
修改Makefile.config文件
- 1
进入caffe目录.
我是将自己之前的caffe的Makefile.config拷贝到了该目录下,也可以执行
- 1
对Makefile.config
文件进行修改.
注意修改
- 1
- 2
- 3
修改cudnn加速文件
这里需要注意一下,因为作者的caffe版本不支持新的cudnn加速,我们需要将
- 1
- 2
- 3
- 4
选择caffe-master版的文件进行替换,这样编译就不会出错啦.
修改一个include错误
修改
- 1
删除第十一行
- 1
编译caffe
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
下载已经训练好的模型
- 1
- 2
执行Demo
- 1
- 2
然后就大功告成了.