我的caffe+GPU1070Ti+CUDA9.0+cuDNN7.0.5在Ubuntu16.04下已经配置好了,我在此参考py-faster-rcnn官网和一些博客对其进行配置的。
1、py-faster-rcnn官网:https://github.com/rbgirshick/py-faster-rcnn
2、Ubuntu16.04 faster-rcnn+caffe+gpu运行环境配置以及解决各种bug:https://blog.csdn.net/flygeda/article/details/78638824
3、ubuntu14.04 + cuda8.0 + cudnnv5 + caffe + py-faster-rcnn配置 :http://www.cnblogs.com/zf-blog/p/6701710.html
根据我的配置过程,我觉得在caffe+GPU1070Ti+CUDA9.0+cuDNN7.0.5或其他已安装好的情况下,最麻烦的可能就是py-faster-rcnn下的caffe-fast-rcnn中对于高于cudnn版本高于4.0不兼容问题解决,需要用最新的caffe里的文件替换掉caffe-fast-rcnn相关文件,网上看到的大致有三种方式:
1、采用手动复制的方式
2、采用git命令行合并的方式
3、采用命令终端使用cp命令行复制方式
不建议使用方式1,因为切换的目录多。如果以前没有安装过caffe,那就采用git命令行联网合并融合;如果已经安装了caffe或下载了caffe那就是用第3中方式好些,因为下载caffe或caffe-fast-rcnn需要连外网,网络一不给力可能就中断了。我前面安装了caffe,所以在此就采用第3种方式,我的caffe放在/home下,py-faster-rcnn放置在/home/Project/,因此我的命令行操作为:
cd
cp caffe/include/caffe/layers/cudnn_* Project/py-faster-rcnn/caffe-fast-rcnn/include/caffe/layers/
cp caffe/src/caffe/layers/cudnn_* Project/py-faster-rcnn/caffe-fast-rcnn/src/caffe/layers/
cp caffe/include/caffe/util/cudnn.hpp Project/py-faster-rcnn/caffe-fast-rcnn/include/caffe/util/
其他的配置参考上面3篇博客就行了。