C++11
CUDA 9.0
Compile and run the train.cu file:
nvcc --std=c++11 train.cu -o train
./train
To execute only on the CPU, change the last argument of the Train() function in train.cu to true.
-
Notifications
You must be signed in to change notification settings - Fork 0
akhandait/CudaNeuralNets
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Parallelizing Neural Networks - Course project for High Performance Scientific Computing
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published