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
번호 | 제목 | 글쓴이 | 날짜 | 조회 수 |
---|---|---|---|---|
1519 | good luck ! | WHRIA | 2019.03.21 | 315 |
1518 | 민원24 프린터제한을 우회하거나 PDF 로 저장하는 방법 | WHRIA | 2015.07.12 | 319 |
1517 | nvidia dali [1] | WHRIA | 2020.08.01 | 328 |
1516 | % 빼고 center crop | WHRIA | 2016.10.22 | 329 |
1515 | IPTV | WHRIA | 2015.11.01 | 332 |
1514 | IEEE Spectrum | WHRIA | 2018.02.27 | 334 |
1513 | reverse VNC | WHRIA | 2018.02.14 | 346 |
1512 | python pdf writer | WHRIA | 2019.11.04 | 349 |
1511 | CNN Models | WHRIA | 2018.12.23 | 353 |
1510 | scopus [1] | WHRIA | 2020.08.02 | 355 |
1509 | Keratinocytic Skin Cancer Detection on the Face Using Region-Based Convolutional Neural Network | WHRIA | 2019.12.07 | 364 |
1508 | 미래가 보이지 않는다. | WHRIA | 2017.07.01 | 367 |
1507 | gpt2 | WHRIA | 2023.02.21 | 398 |
1506 | Windows Subsystem Linux | WHRIA | 2018.01.01 | 425 |
1505 | HASh benchmark | WHRIA | 2019.03.10 | 436 |