Peter9606 / neuralnetwork

A simple neural network implemented by using cudnn and cublas

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

About

A simple neural network implemented by using cudnn and cublas

License:Apache License 2.0


Languages

Language:C++ 91.2%Language:CMake 6.2%Language:Cuda 2.5%