看到这里,默认已经安装并编译过caffe了。
要学习caffe源码就要调试,在windows下调试。使用visual studio 2015调试caffe源码。
第一步:将生成caffe.lib的caffe项目设置为启动项。
第二步:以mnist为例调试caffe,右键caffe项目,进入属性设置。
在命令处填上可执行程序路径,在命令参数行填上可执行程序需要的参数。
第三步:下面就是具体调试了,自己跟代码,不赘述。
特别说明:在使用GPU调试时可能会报错:
F0928 15:22:49.693164 12016 cudnn.hpp:124] Check failed: status == CUDNN_STATUS_SUCCESS (3 vs. 0) CUDNN_STATUS_BAD_PARAM
caffe-d.exe has triggered a breakpoint.
在网上查了一下解决办法:
对caffe/examples/mnist/lenet_train_test.prototxt,在convolution_param中添加engine:caffe
一共有两处,如下图: