Skip to content

fast rcnn cudnn 5.0

2017.01.08 15:47

WHRIA 조회 수:128

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

번호 제목 글쓴이 날짜 조회 수
1611 melanoma awareness [1] WHRIA 2019.12.27 45
1610 윈도우 raid ahci 전환 WHRIA 2019.12.26 37
1609 lvm 확장 [1] WHRIA 2019.12.25 48
1608 pytorch object detect / retinanet WHRIA 2019.12.22 214
1607 kaggle leakage WHRIA 2019.12.19 35
1606 pytorch - caffe WHRIA 2019.12.19 37
1605 pytorch mini batch size WHRIA 2019.12.19 82
1604 windows softraid monitor WHRIA 2019.12.19 66
1603 raid monitor WHRIA 2019.12.15 31
1602 SAS controller SAS 9212-4i WHRIA 2019.12.08 143
1601 Basic tutorial to develop driver on windows WHRIA 2019.12.07 375
1600 Keratinocytic Skin Cancer Detection on the Face Using Region-Based Convolutional Neural Network WHRIA 2019.12.07 324
1599 linux raid WHRIA 2019.11.30 46
1598 Micro- and macro-averages WHRIA 2019.11.28 140
1597 mcnemar WHRIA 2019.11.20 76

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


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

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

설치 취소