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
번호 | 제목 | 글쓴이 | 날짜 | 조회 수 |
---|---|---|---|---|
1671 | git 특정 버젼 | WHRIA | 2020.07.11 | 34 |
1670 | naver rexnet clova AI [1] | WHRIA | 2020.07.04 | 229 |
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 | 3224 |
1657 | 영어로 숫자 표현 | WHRIA | 2020.05.28 | 4447 |