http://blog.csdn.net/u011635764/article/details/52831167
http://blog.csdn.net/u012841667/article/details/53436615
安装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
然后就大功告成了.
如遇到 这个问题 make: *** [.build_release/src/caffe/layers/hdf5_output_layer.o] Error 1
修改Makefile.config需要修改的内容:
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