The GPU included on the system is a K520 with 4GB of memory and 1,536 cores. This instance is named the g2.2xlarge instance and costs approximately $0.65 per hour. How to install CUDA Toolkit and cuDNN for deep learningĪs I mentioned in an earlier blog post, Amazon offers an EC2 instance that provides access to the GPU for computation purposes. Feel free to spin up an instance of your own and follow along.īy the time you’re finished this tutorial, you’ll have a brand new system ready for deep learning. Specifically, I’ll be using an Amazon EC2 g2.2xlarge machine running Ubuntu 14.04. In the remainder of this blog post, I’ll demonstrate how to install both the NVIDIA CUDA Toolkit and the cuDNN library for deep learning. Using the cuDNN package, you can increase training speeds by upwards of 44%, with over 6x speedups in Torch and Caffe.
0 Comments
Leave a Reply. |