/DIGITS

Deep Learning GPU Training System

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

DIGITS

Build Status Coverage Status Code Health

DIGITS (the Deep Learning GPU Training System) is is a webapp for training deep learning models.

Table of Contents

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 depends on several third-party libraries.

CUDA

The CUDA toolkit (>= CUDA 6.5) is highly recommended, though not technically required.

  • Download from the CUDA website
  • Follow the installation instructions
  • Don't forget to set your path. For example:
    • CUDA_HOME=/usr/local/cuda
    • LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

For greater performance, you can also install cuDNN.

  • Download from the cuDNN website
  • Follow the installation instructions
  • You can copy the files directly into your CUDA installation
    • cp -a libcudnn* $CUDA_HOME/lib64/
    • cp cudnn.h $CUDA_HOME/include/

Deep learning frameworks

At least one deep learning framework backend is required.

Install DIGITS

Grab the source

% cd $HOME
% git clone https://github.com/NVIDIA/DIGITS.git digits

Throughout the docs, we'll refer to your install location as DIGITS_HOME ($HOME/digits in this case), though you don't need to actually set that environment variable.

Python dependencies

Several PyPI packages need to be installed. The recommended installation method is using a virtual environment (installation instructions).

% cd $DIGITS_HOME
% pip install -r requirements.txt

If you want to install these packages without using a virtual environment, replace "pip install" with "sudo pip install".

Other dependencies

DIGITS uses graphviz to visualize network architectures. You can safely skip this step if you don't want the feature.

% sudo apt-get install graphviz

Starting the server

You can run DIGITS in two ways:

Development mode

% ./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).

Then, you can view your server at http://localhost:5000/.

Production mode

% ./digits-server

Starts a production server (gunicorn) that listens on port 8080 (http://localhost:8080). If you get any errors about an invalid configuration, use the development server first to set your configuration.

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.