Skip to content

fast rcnn cudnn 5.0

2017.01.08 15:47

WHRIA 조회 수:131

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

번호 제목 글쓴이 날짜 조회 수
1624 오랜만에 글을 남김 WHRIA 2012.01.18 6875
1623 법인 설립 WHRIA 2008.10.04 6820
1622 viber 를 써보니 WHRIA 2012.01.23 6798
1621 미남계 미녀계 WHRIA 2007.07.06 6785
1620 온몸이 쑤심 WHRIA 2007.07.01 6748
1619 spirit WHRIA 2007.08.27 6721
1618 지갑 바꾸다. WHRIA 2007.07.17 6691
1617 Medieye Release 2.0.8.7 [1] WHRIA 2007.08.16 6679
1616 배구반 WHRIA 2007.07.02 6671
1615 Let's get start WHRIA 2007.08.06 6669
1614 선택과 집중 WHRIA 2007.06.24 6654
1613 오늘부터 조금씩 보인다. WHRIA 2007.07.29 6559
1612 Dream and Success WHRIA 2007.08.03 6499
1611 우울 WHRIA 2007.07.03 6496
1610 modprobe WHRIA 2020.06.29 6433

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


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

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

설치 취소