/DIGITS

Deep Learning GPU Training System

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

DIGITS (Deep GPU Training System)

DIGITS is is a webapp for training deep learning models.

Get help

Installation issues

  • First, check out the instructions below
  • Then, ask questions on our user group

Usage questions

Bugs and feature requests

Installation

DIGITS is only officially supported on Ubuntu 14.04. However, DIGITS has been successfully used on other Linux variants as well as on OSX.

Prerequisites

DIGITS has several dependencies.

  • CUDA
  • cuDNN library
  • Caffe – NVIDIA branch (version 0.11.0)
  • Python packages
  • Graphviz

CUDA (>= 6.5)

Download from the CUDA website and follow the installation instructions.

cuDNN (>= v2)

Download from the cuDNN website and follow the installation instructions.

NVIDIA's fork of Caffe (NVIDIA version 0.11.0)

Detailed installation instructions are available on caffe's installation page. Condensed version is as follows:

Installing caffe prerequisites on Linux

% sudo apt-get install git
% cd $HOME
% git clone --branch v0.11.0 https://github.com/NVIDIA/caffe.git
% cd caffe
% sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler libatlas-base-dev
% sudo apt-get install python-dev python-pip gfortran
% cd python
% for req in $(cat requirements.txt); do sudo pip install $req; done

Installing caffe prerequisites on Mac OS

If you have homebrew installed, you can follow the instructions from caffe's installation page

Build caffe:

% cd $HOME/caffe
% cp Makefile.config.example Makefile.config
% make all --jobs=4
% make py

Set environment variables:

% export CAFFE_HOME=${HOME}/caffe

Install DIGITS

% cd $HOME
% git clone https://github.com/NVIDIA/DIGITS.git digits
% cd digits
% sudo apt-get install graphviz
% for req in $(cat requirements.txt); do sudo pip install $req; done

Again if on MacOS with homebrew, replace the last two lines as follows:

$ brew install graphviz
$ for req in $(cat requirements.txt); do pip install $req; done

Starting the server

You can run DIGITS in two ways:

digits-devserver

Starts a development server that listens on port 5000 (but you can change the port if you like - try running it with the --help flag).

digits-server

Starts a gunicorn app that listens on port 8080. If you have installed the nginx.site to your nginx sites-enabled/ directory, then you can view your app at http://localhost/.

Usage

Check out the Getting Started page.