Setup DIGITS (Deep Learning GPU Training System) on Ubuntu 16.04

DIGITS (the Deep Learning GPU Training System) is a webapp for training deep learning models. The currently supported frameworks are: Caffe, Torch, and Tensorflow.

Since DIGITS itself is a pure Python project, installation is usually pretty trivial regardless of the platform. The difficulty comes from installing all the required dependencies for Caffe and/or Torch7 and configuring the builds.

Follow below links for installing Caffe and Torch7:

Install dependencies

sudo apt-get install --no-install-recommends git graphviz python-dev python-flask python-gevent python-h5py python-numpy python-pil

Set DIGITS enviornment variable, so edit bashrc and restart Terminal.
gedit ~/.bashrc    #append following lines to it

#Digit Path set
export DIGITS_ROOT=~/digits

Download Source

git clone $DIGITS_ROOT

Python Requirement Package installation:

sudo pip install -r $DIGITS_ROOT/requirements.txt

Install DIGITS and Enable support for plug-ins (You can skip this step if you want to keep setup portable and dont want to use plugins)

DIGITS needs to be installed to enable loading data and visualization plug-ins:

sudo pip install -e $DIGITS_ROOT

Starting the server



#restart terminal for the Digits setup to take effect and run following directly if installed

Starts a server at http://localhost:5000/

so open your browser and open http://localhost:5000 link to open DIGITS Platform


How to Uninstall digits which is installed using easy install or sudo pip install -e flag:

First you have to run this command, which will remove all dependencies of the package.

sudo easy_install -mxN digits

Then remove egg file of that package:

sudo rm -rf /usr/local/lib/python2.7/dist-packages/digits*

This will uninstall digits for you.









One thought on “Setup DIGITS (Deep Learning GPU Training System) on Ubuntu 16.04

Add yours

Join the Discussion

Fill in your details below or click an icon to log in: Logo

You are commenting using your account. Log Out /  Change )

Google photo

You are commenting using your Google account. Log Out /  Change )

Twitter picture

You are commenting using your Twitter account. Log Out /  Change )

Facebook photo

You are commenting using your Facebook account. Log Out /  Change )

Connecting to %s

Create a free website or blog at

Up ↑

%d bloggers like this: