fast rcnn cudnn 5.0
2017.01.08 15:47
手动修改文件,参考了卜居大神的博客 http://blog.csdn.net/kkk584520/article/details/51163564方案1简单方便,但是当我们编译的是其他人修改过得源码,可能就会出错。方案2 步骤如下:
1. 用最新caffe源码的以下文件替换掉faster rcnn 的对应文件
include/caffe/layers/cudnn_relu_layer.hpp, src/caffe/layers/cudnn_relu_layer.cpp, src/caffe/layers/cudnn_relu_layer.cu
include/caffe/layers/cudnn_sigmoid_layer.hpp, src/caffe/layers/cudnn_sigmoid_layer.cpp, src/caffe/layers/cudnn_sigmoid_layer.cu
include/caffe/layers/cudnn_tanh_layer.hpp, src/caffe/layers/cudnn_tanh_layer.cpp, src/caffe/layers/cudnn_tanh_layer.cu
2. 用caffe源码中的这个文件替换掉faster rcnn 对应文件
include/caffe/util/cudnn.hpp
3. 将 faster rcnn 中的 src/caffe/layers/cudnn_conv_layer.cu 文件中的所有
cudnnConvolutionBackwardData_v3 函数名替换为 cudnnConvolutionBackwardData
cudnnConvolutionBackwardFilter_v3函数名替换为 cudnnConvolutionBackwardFilter
댓글 0
번호 | 제목 | 글쓴이 | 날짜 | 조회 수 |
---|---|---|---|---|
216 | 넌센스 퀴즈 | WHRIA | 2016.09.04 | 75 |
215 | 공유 메모리 예제 | WHRIA | 2016.09.09 | 121 |
214 | 윈도우 서비스에서 유저 권한으로 createprocess | WHRIA | 2016.09.10 | 1995 |
213 | 힘든 하루하루 | WHRIA | 2016.09.30 | 77 |
212 | youtube | WHRIA | 2016.09.30 | 39 |
211 | 만기까지 끌어 올림. | WHRIA | 2016.09.30 | 70 |
210 | . | WHRIA | 2016.10.01 | 81 |
209 | KMA 연수교육 주소 | WHRIA | 2016.10.05 | 100 |
208 | sqlite to mysql | WHRIA | 2016.10.18 | 57 |
207 | caffe | WHRIA | 2016.10.19 | 106 |
206 | 이미지 size 줄이는 batch | WHRIA | 2016.10.19 | 71 |
205 | 이미지를 정사각형으로 만드는 batch | WHRIA | 2016.10.22 | 1177 |
204 | % 빼고 center crop | WHRIA | 2016.10.22 | 297 |
203 | jquery java | WHRIA | 2016.12.13 | 1757 |
» | fast rcnn cudnn 5.0 | WHRIA | 2017.01.08 | 128 |