faster_rcnn icon indicating copy to clipboard operation
faster_rcnn copied to clipboard

Support cudnn on windows?

Open 7oud opened this issue 8 years ago • 4 comments

Whether the faster rcnn support cudnn on windows version? And if so which version of cudnn can be used?

7oud avatar Mar 18 '16 08:03 7oud

yes, i'm using it. The gpu compute capability needs to be higher than 3.0, the version of cudnn doesn't matter. I'm using cudnn-7.5-windows7-x64-v5.1.

lixiang-ucas avatar Aug 31 '16 14:08 lixiang-ucas

using cudnn will cause the LRN layer so slow, please change the engine in the prototxt

xiaoxiongli avatar Sep 12 '16 05:09 xiaoxiongli

how to change the engine in prototxt, i cannot find the parameters indicate whether to use cudnn.

lixiang-ucas avatar Sep 12 '16 07:09 lixiang-ucas

@lixiang-ucas it is very easy to control every layer's "engine", just follow this: layer { name: "norm1" type: "LRN" bottom: "conv1" top: "norm1" lrn_param { local_size: 3 alpha: 0.00005 beta: 0.75 norm_region: WITHIN_CHANNEL engine: CAFFE } }

now we have below 3 choice, the layer's default value is DEFAULT:

DEFAULT maybe means that if you compile caffe with CUDNN support, then the DEFAULT==CUDNN, otherwise, the DEFAULT==CAFFE.

enum Engine { DEFAULT = 0; CAFFE = 1; CUDNN = 2; }

xiaoxiongli avatar Sep 23 '16 08:09 xiaoxiongli