Deep Neural Network Framework in CUDA

Hey guys, long time no see! I’m happy to show you the project I’m working on recently. I transplanted my convolutional neural network implementation into GPU environment, and made a deep neural network framework in nVidia CUDA. Although it is not completely finished, most of the functions are available for use. You are more than welcome to try and play with it, and I appreciate if you could contribute or tell me any bug of it.

https://github.com/xingdi-eric-yuan/cuda-deep-neural-nets

Since I’m still using my 2011 Macbook Pro (it’s my only computer which has NVIDIA graph card), the running speed of that code on my computer is fairly slow, but I’m trying to optimize it:)

111

 

This entry was posted in Machine Learning, Twaddle and tagged , . Bookmark the permalink. Post a comment or leave a trackback: Trackback URL.

3 Comments

  1. Posted August 14, 2015 at 10:56 pm | Permalink

    Hi Eric.

    Could this be used in a demo similar to the Deep mind atari demo?

    e.g http://cs.stanford.edu/people/karpathy/convnetjs/demo/rldemo.html?

    Could you use cuDNN?

    https://developer.nvidia.com/cudnn

  2. Yu Deng
    Posted October 13, 2015 at 9:06 pm | Permalink

    Hi ERIC, Thank you for your excellent work. Can I build the project using visual studio 2010? Sincerely yours
    Yu Deng

  3. iamhuskar
    Posted December 22, 2016 at 10:22 pm | Permalink

    感谢博主的博客。满满都是机器学习的干货。受益匪浅。准备从入门看起了。想想本科数学专业,却没有认真学习真实后悔。现在只记得几个人名和基本的名词。又要把知识捡起来了~

Post a Comment

Your email is never published nor shared. Required fields are marked *

You may use these HTML tags and attributes <a href="" title=""> <abbr title=""> <acronym title=""> <b> <blockquote cite=""> <cite> <code> <del datetime=""> <em> <i> <q cite=""> <s> <strike> <strong>

*
*