add support for cudnn6 - Githubissues

TimoSaemann / caffe-segnet-cudnn5

This repository was a fork of BVLC/caffe and includes the upsample, bn, dense_image_data and softmax_with_loss (with class weighting) layers of caffe-segnet (https://github.com/alexgkendall/caffe-segnet) to run SegNet with cuDNN version 5.

Other

176 stars 128 forks source link

add support for cudnn6 #20

Open kertansul opened 6 years ago

kertansul commented 6 years ago

Verified in cudnn6

Ref: https://github.com/BVLC/caffe/commit/cd2a0140d60742d4f66cc1668a2554bdc7a1ca3c

hzxie commented 6 years ago

Verified in cudnn7