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
번호 | 제목 | 글쓴이 | 날짜 | 조회 수 |
---|---|---|---|---|
1519 | JAMA | WHRIA | 2017.02.11 | 674 |
1518 | windows faster rcnn | WHRIA | 2017.02.10 | 4703 |
1517 | inception | WHRIA | 2017.01.23 | 7205 |
1516 | fast rcnn | WHRIA | 2017.01.08 | 133 |
» | fast rcnn cudnn 5.0 | WHRIA | 2017.01.08 | 161 |
1514 | jquery java | WHRIA | 2016.12.13 | 1799 |
1513 | % 빼고 center crop | WHRIA | 2016.10.22 | 330 |
1512 | 이미지를 정사각형으로 만드는 batch | WHRIA | 2016.10.22 | 1203 |
1511 | 이미지 size 줄이는 batch | WHRIA | 2016.10.19 | 116 |
1510 | caffe | WHRIA | 2016.10.19 | 127 |
1509 | sqlite to mysql | WHRIA | 2016.10.18 | 111 |
1508 | KMA 연수교육 주소 | WHRIA | 2016.10.05 | 129 |
1507 | . | WHRIA | 2016.10.01 | 102 |
1506 | 만기까지 끌어 올림. | WHRIA | 2016.09.30 | 88 |
1505 | youtube | WHRIA | 2016.09.30 | 65 |