Skip to content

atomwalk12/neural-network-cuda

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

48 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Running the code

Run the training script on the CPU and GPU:

> nvcc CPU/main.cpp CPU/linear.cpp CPU/relu.cpp CPU/train.cpp data/read_csv.cpp CPU/sequential.cpp CPU/mse.cpp utils/utils.cpp -o cpu_trainer
> nvcc GPU/main.cu GPU/linear.cu GPU/relu.cu GPU/train.cu data/read_csv.cpp GPU/sequential.cu utils/utils.cpp GPU/mse.cu -o gpu_trainer

Reference Papers:

About

Neural network from scratch in CUDA/C++

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Cuda 56.6%
  • C++ 40.8%
  • C 2.6%