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:
sudo apt-get install --no-install-recommends git graphviz python-dev python-flask python-flaskext.wtf 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
git clone https://github.com/NVIDIA/DIGITS.git $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
cd $DIGITS_ROOT ./digits-devserver
#restart terminal for the Digits setup to take effect and run following directly if installed digits-devserver
Starts a server at
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.