alexgkendall / SegNet-Tutorial

Files for a tutorial to train SegNet for road scenes using the CamVid dataset
http://mi.eng.cam.ac.uk/projects/segnet/tutorial.html
851 stars 518 forks source link

cuda8+cudnn v5 for Segnet #58

Open guhaohitsz opened 8 years ago

guhaohitsz commented 8 years ago

Recently, i have change to use GTX1080, so i install cuda8 and the respectively cudnn v5. how can i change the code for my computer? Regards!

Timo-hab commented 8 years ago

Are you progressed at it? Did you try it with cudnn v3 as well?

aleksandrmelnyk commented 7 years ago

cudnn v3 works cudnn v5 doesn't work

would be great if segnet is upgraded to latest caffe.

Timo-hab commented 7 years ago

https://github.com/TimoSaemann/caffe-segnet-cudnn5