MehdiNoroozi/JigsawPuzzleSolver

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_*