Skip to content

fast rcnn cudnn 5.0

2017.01.08 15:47

WHRIA 조회 수:157

手动修改文件,参考了卜居大神的博客 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

번호 제목 글쓴이 날짜 조회 수
259 Micro- and macro-averages WHRIA 2019.11.28 152
258 pytorch optimize WHRIA 2020.08.01 152
257 kakao WHRIA 2018.02.25 149
256 모든 code 를 python3 로 migration 중 WHRIA 2019.06.08 148
255 model split [1] WHRIA 2020.07.30 146
254 이궁 WHRIA 2019.03.12 143
253 Visual Studio 필요없는 파일 삭제 for SVN file WHRIA 2016.09.04 138
252 나이 40에... WHRIA 2018.09.01 138
251 indexeddb [2] WHRIA 2021.02.07 138
250 file lock WHRIA 2020.09.22 137
249 공유 메모리 예제 WHRIA 2016.09.09 136
248 은행은 믿을 곳이 못 되는군... WHRIA 2019.07.12 136
247 fast rcnn WHRIA 2017.01.08 133
246 벤츠 E300 2010 년형 터치스크린 자가 수리 WHRIA 2015.12.25 132
245 PHP7 으로 업그레이드 WHRIA 2016.05.05 129

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


이 PC에는 나눔글꼴이 설치되어 있지 않습니다.

이 사이트를 나눔글꼴로 보기 위해서는
나눔글꼴을 설치해야 합니다.

설치 취소