Thunder icon indicating copy to clipboard operation
Thunder copied to clipboard

A small deep-learning framework with C++/Python/CUDA

Thunder

A small deep-learning framework with C++/Python/CUDA

一个微小的深度学习框架,使用了C++/Python/CUDA

参考了

  • Pytorch
  • MXnet
  • Tensorflow
  • Auora

使用方法

git clone https://github.com/HadXu/Thunder.git
cd cuda
make

进入 examples

python mnist

欢迎提出意见