ubuntu16.04中gpu版本的caffe-ssd模型训练与测试

最近把一个ssd网络的net..prototxt网络结构和自己生成的hdf5格式数据一起做训练时发现经常报错，因为ssd中一些层在caffe中并没有实现，需要自己写相应的.cpp,.cu文件重新编译，比较麻烦，而大家通常训练caffe-ssd都是基于原作者公开的代码训练的，该代码中实现了这些层，于是把原作者代码跑了一遍．

我的机器是ubuntu16.04,装有两块ＧＴＸ　１０８０TI 的显卡，一共２０多个Ｇ显存，opencv是自己编译的3..3.0版本．

因为我机器主目录下已经有个装好的caffe了，所以再次在另一个目录下载该源码，该源码下载后编译也能import caffe，其他caffe项目也可以使用，如果没有装caffe，那就下载编译这个源码就能当做caffe使用，同时也能训练ssd．

１，先进入我这次操作的主目录（/home/user/yjf),下载源码

git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd（切换到ｓｓｄ分支）

这时候我的工作目录为 /home/user/yjf/caffe．然后执行如下操作

cp Makefile.config.example Makefile.config

然后参考我前面博客说过的如何搭建caffe环境文章中的说法更改Makefile和Makefile.config文件，并参照里面提到的方法进行编译（我感觉caffe不管是ｃｐｕ还是ｇｐｕ编译起来其实很简单的，我一般３０分钟就能弄好，没有大家想象中的有难度或者