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
번호 | 제목 | 글쓴이 | 날짜 | 조회 수 |
---|---|---|---|---|
319 | 넌센스 퀴즈 | WHRIA | 2016.09.04 | 82 |
318 | 공유 메모리 예제 | WHRIA | 2016.09.09 | 138 |
317 | 윈도우 서비스에서 유저 권한으로 createprocess | WHRIA | 2016.09.10 | 2019 |
316 | 힘든 하루하루 | WHRIA | 2016.09.30 | 83 |
315 | youtube | WHRIA | 2016.09.30 | 61 |
314 | 만기까지 끌어 올림. | WHRIA | 2016.09.30 | 85 |
313 | . | WHRIA | 2016.10.01 | 102 |
312 | KMA 연수교육 주소 | WHRIA | 2016.10.05 | 123 |
311 | sqlite to mysql | WHRIA | 2016.10.18 | 103 |
310 | caffe | WHRIA | 2016.10.19 | 123 |
309 | 이미지 size 줄이는 batch | WHRIA | 2016.10.19 | 110 |
308 | 이미지를 정사각형으로 만드는 batch | WHRIA | 2016.10.22 | 1199 |
307 | % 빼고 center crop | WHRIA | 2016.10.22 | 329 |
306 | jquery java | WHRIA | 2016.12.13 | 1792 |
» | fast rcnn cudnn 5.0 | WHRIA | 2017.01.08 | 157 |