https://github.com/NVIDIA/DIGITS
Raw File
Tip revision: ed022de723a5c71676dc2ba5487c1334c9725b53 authored by Luke Yeager on 06 July 2015, 18:33:44 UTC
Mark v2.0.0-preview
Tip revision: ed022de
README.md
# DIGITS

[![Build Status](https://travis-ci.org/NVIDIA/DIGITS.svg?branch=master)](https://travis-ci.org/NVIDIA/DIGITS)
[![Coverage Status](https://coveralls.io/repos/NVIDIA/DIGITS/badge.svg?branch=master)](https://coveralls.io/r/NVIDIA/DIGITS?branch=master)
[![Code Health](https://landscape.io/github/NVIDIA/DIGITS/master/landscape.svg?style=flat)](https://landscape.io/github/NVIDIA/DIGITS/master)

DIGITS (the **D**eep Learning **G**PU **T**raining **S**ystem) is is a webapp for training deep learning models.

Table of Contents
=================
* [Get help](#get-help)
* [Installation](#installation)
  * [Prerequisites](#prerequisites)
  * [Install DIGITS](#install-digits)
* [Starting the server](#starting-the-server)

# Get help

### Installation issues
* First, check out the instructions below
* Then, ask questions on our [user group](https://groups.google.com/d/forum/digits-users)

### Usage questions
* First, check out the [Getting Started](docs/GettingStarted.md) page
* Then, ask questions on our [user group](https://groups.google.com/d/forum/digits-users)

### Bugs and feature requests
* Please let us know by [filing a new issue](https://github.com/NVIDIA/DIGITS/issues/new)
* Bonus points if you want to contribute by opening a [pull request](https://help.github.com/articles/using-pull-requests/)!
  * You will need to send a signed copy of the [Contributor License Agreement](CLA) to digits [at] nvidia [dot] com before your change can be accepted.

# 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](https://developer.nvidia.com/cuda-downloads)
* 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](https://developer.nvidia.com/cuDNN)
* 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.

* Caffe (NVIDIA's fork) - [installation instructions](docs/InstallCaffe.md)
* Torch7 - *coming soon*

## 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](docs/VirtualEnvironment.md)).

    % 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](docs/GettingStarted.md) page.
back to top