fastalgo/cudnn-training

A CUDNN minimal deep learning training code sample using LeNet.

A CUDNN minimal deep learning training code sample using LeNet.

Prerequisites

• 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 (v5 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.

Compilation

The project can either be compiled with CMake (cross-platform) or Visual Studio.

To compile with CMake, run the following commands:

~: $ cd cudnn-training/
~/cudnn-training: $ mkdir build
~/cudnn-training: $ cd build/
~/cudnn-training/build: $ cmake ..
~/cudnn-training/build: $ make

If compiling under linux, make sure to either set the CUDNN_PATH environment variable to the path CUDNN is installed to, or extract CUDNN to the CUDA toolkit path.

To enable gflags support, uncomment the line in CMakeLists.txt. In the Visual Studio project, define the macro USE_GFLAGS.

Running

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.