search

willyd / caffe-builder

CMake build scripts to automate building the Caffe library and its dependencies.
BSD 2-Clause "Simplified" License
186 stars 116 forks source link

Is CuDNN support enabled by default? #24

Closed mrgloom closed 8 years ago

mrgloom commented 8 years ago

Is CuDNN support enabled by default? How I can check if caffe compiled with CuDNN?

willyd commented 8 years ago

No since you need to be a registered NVIDIA developper to have access to cuDNN. You need to use the msvc branch of my caffe repo to use cuDNN. See https://github.com/willyd/caffe/blob/msvc/cmake/Cuda.cmake#L174 to have an idea on how to enable cuDNN, i.e. what CMake variables need to be set.