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

번호 제목 글쓴이 날짜 조회 수
1671 git 특정 버젼 WHRIA 2020.07.11 34
1670 naver rexnet clova AI [1] WHRIA 2020.07.04 228
1669 sql python WHRIA 2020.07.01 180
1668 modprobe WHRIA 2020.06.29 6432
1667 tab 구조 html WHRIA 2020.06.28 79
1666 imagenet classification SOTA WHRIA 2020.06.27 1398
1665 할일 WHRIA 2020.06.24 1464
1664 caffe hdf5 error WHRIA 2020.06.22 187
1663 resnest50 caffe WHRIA 2020.06.21 163
1662 reddit downloader WHRIA 2020.06.21 204
1661 바이두 WHRIA 2020.06.20 627
1660 regression model [6] WHRIA 2020.06.18 167
1659 MS bot framework [3] WHRIA 2020.06.13 99
1658 BERT WHRIA 2020.06.10 3223
1657 영어로 숫자 표현 WHRIA 2020.05.28 4447

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


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

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

설치 취소