NVIDIA DIGITS with Caffe – Performance on Pascal multi-GPU

NVIDIA’s Pascal GPU’s have twice the computational performance of the last generation. A great use for this compute capability is for training deep neural networks. We have tested NVIDIA DIGITS 4 with Caffe on 1 to 4 Titan X and GTX 1070 cards. Training was for classification of a million image data set from ImageNet. Read on to see how it went.

NVIDIA DIGITS Install

If you are happy to use Ubuntu 14.04 LTS (Ubuntu-MATE in our case) then setting up a system with the NVIDIA DIGITS software stack is simple. I’ll give you some guidance on getting everything working, from the Linux install to the DIGITS web interface.

What is Machine Learning

Machine Learning is getting a lot of attention these days and with good reason. There are mountains of data to work with and computing resources to handle the problems are easily attainable. Even a single GPU accelerated workstation is capable of serious work.