By default do not choose CUDNN engine in CPU mode

BVLC / caffe

Caffe: a fast open framework for deep learning.

http://caffe.berkeleyvision.org/

Other

34.09k stars 18.7k forks source link

Open dashesy opened 5 years ago

dashesy commented 5 years ago

even if it is compiled with USE_CUDNN

Fixes #3317

artyom-beilis commented 5 years ago

There is an up-to-date pool request that fixes it: https://github.com/BVLC/caffe/pull/6775

yezhengli-Mr9 commented 3 years ago

even if it is compiled with USE_CUDNN

Fixes #3317

Any tutorial to switch to CPU-only caffe?caffe.set_mode_cpu() seems not enough. I am studying another problem anyway.