/neuralnetwork

A simple neural network implemented by using cudnn and cublas

Primary LanguageC++Apache License 2.0Apache-2.0

Neural Network in cuDNN

This is a neural network written mainly in cuDNN, as well as cuBLAS. It is NOT reinventing the wheel, it's to help people who just started cuda toolkits to easily twist with so that fast learn cudnn and cublas ect.