What are the cuda, cudnn and caffe version used
Opened this issue · 1 comments
Hi,
Please can you update the cuda, cudnn and caffe version used ?
I am getting different kind of errors while trying to compile with different cudnn versions.
cd caffe-maste-jps
make clean all
It would be great if you can tell me the source files changed for customized caffe (that way I can use the latest cudnn version which wiil be faster) ?
Hi,
Sorry for late reply. I saw your message just before. We have trained a new version using stride 4 in transfer learning and got much better results. Will release in Aug.
I have changed several files. The faster way would be to compile my branch with the new cudnn. You need to replace the following files with those in a version of caffe that works in your machine.
include/caffe/util/cudnn.hpp
include/caffe/layers/cudnn_*
src/caffe/util/cudnn.cpp
src/caffe/layers/cudnn_*