search

pengxj / action-faster-rcnn

59 stars 22 forks source link

cudnn v5 support #2

Closed vra closed 2 years ago

vra commented 7 years ago

Hi @pengxj , It seems that the cudnn v5 is not supported in this project, When I make with cudnn v5, it reports errors, while it's fine using cudnn v4. Could you please update the upstream of caffe to make it support cudnn v5? thanks.

pengxj commented 7 years ago

Hi @vra , Hmm. I guess you can simply clone the latest version of caffe. And then change the python interface path in action_tools/init.py. If it's still the problem, let me know.

vra commented 7 years ago

Hi @pengxj , Thanks for your response, I will try it this way.

pengxj commented 7 years ago

My bad. You should clone the caffe tool from https://github.com/rbgirshick/py-faster-rcnn. The official caffe doesnot support ROI pooling.

vra commented 7 years ago

Okay got it

vguptai commented 6 years ago

@vra You can try this also. Hope this helps.

https://medium.com/@vikram.gupta/fixing-py-faster-rcnn-for-cudnn5-a7993d3ff3a1