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
번호 | 제목 | 글쓴이 | 날짜 | 조회 수 |
---|---|---|---|---|
1701 | CORS | WHRIA | 2021.01.22 | 133 |
1700 | popup | WHRIA | 2020.12.31 | 99 |
1699 | bay trail ubuntu 20.04 [1] | WHRIA | 2020.12.24 | 57 |
1698 | 32 bit UEFI [6] | WHRIA | 2020.12.24 | 205 |
1697 | sroc [1] | WHRIA | 2020.12.10 | 61 |
1696 | test | WHRIA | 2020.12.03 | 47 |
1695 | dkms for r8125 | WHRIA | 2020.11.12 | 96 |
1694 | unattended upgrade | WHRIA | 2020.11.01 | 478 |
1693 | pytorch pretrained | WHRIA | 2020.10.28 | 150 |
1692 | steamlit | WHRIA | 2020.10.15 | 211 |
1691 | sample size | WHRIA | 2020.10.13 | 4235 |
1690 | Transformer | WHRIA | 2020.10.09 | 155 |
1689 | file lock | WHRIA | 2020.09.22 | 99 |
1688 | onnx broswer | WHRIA | 2020.09.15 | 42 |
1687 | fda 인증 | WHRIA | 2020.09.03 | 213 |