A CUDNN minimal deep learning training code sample using LeNet.
- C++11 capable compiler (Visual Studio 2013, GCC 4.9 etc.) (for chrono and random)
- CUDA (6.5 or newer): https://developer.nvidia.com/cuda-downloads
- CUDNN (v2 or newer): https://developer.nvidia.com/cuDNN/
- MNIST dataset: http://yann.lecun.com/exdb/mnist/
- (optional) gflags: https://github.com/gflags/gflags
Set the CUDNN_PATH environment variable to where CUDNN is installed.
The project can either be compiled with CMake (cross-platform) or Visual Studio.
To enable gflags support, uncomment the line in CMakeLists.txt. In the Visual Studio project, define the macro USE_GFLAGS.
Extract the MNIST training and test set files (*-ubyte) to a directory (if gflags are not used, the default is the current path).
You can also use the pre-trained weights published along with CUDNN, using the "pretrained" flag.