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
번호 | 제목 | 글쓴이 | 날짜 | 조회 수 |
---|---|---|---|---|
1686 | ubuntu cuda nvidia-smi | WHRIA | 2020.08.29 | 850 |
1685 | concat network [3] | WHRIA | 2020.08.27 | 147 |
1684 | GPT2 [1] | WHRIA | 2020.08.03 | 250 |
1683 | scopus [1] | WHRIA | 2020.08.02 | 344 |
1682 | melafind | WHRIA | 2020.08.01 | 771 |
1681 | nvidia dali [1] | WHRIA | 2020.08.01 | 265 |
1680 | pytorch optimize | WHRIA | 2020.08.01 | 126 |
1679 | startup , 미국 | WHRIA | 2020.07.30 | 134 |
1678 | consort , stard | WHRIA | 2020.07.30 | 81 |
1677 | model split [1] | WHRIA | 2020.07.30 | 142 |
1676 | FTC [1] | WHRIA | 2020.07.28 | 238 |
1675 | 암호화 | WHRIA | 2020.07.26 | 1000 |
1674 | raid 6 | WHRIA | 2020.07.23 | 43 |
1673 | asyncio | WHRIA | 2020.07.23 | 286 |
1672 | amp distributed pytorch [1] | WHRIA | 2020.07.14 | 51 |