DIGITS is is a webapp for training deep learning models.
- First, check out the instructions below
- Then, ask questions on our user group
- First, check out the Getting Started page
- Then, ask questions on our user group
- Please let us know by filing a new issue
- Bonus points if you want to contribute by opening a pull request!
- You will need to send a signed copy of the Contributor License Agreement to digits [at] nvidia [dot] com before your change can be accepted.
DIGITS is only officially supported on Ubuntu 14.04. However, DIGITS has been successfully used on other Linux variants as well as on OSX.
DIGITS depends on several third-party libraries.
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/
At least one deep learning framework backend is required.
- Caffe (NVIDIA's fork) - installation instructions
- Torch7 - coming soon
% 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.
These packages need to be installed:
% cd $DIGITS_HOME
% cat requirements.txt
The recommended installation method is using a virtual environment (installation instructions).
You may be tempted to install the requirements with pip install -r requirements.txt
. Use the loop statement below to install the packages in order and avoid errors.
% for req in $(cat requirements.txt); do pip install $req; done
If you want to install these packages without using a virtual environment, replace "pip install" with "sudo pip install".
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
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).
Then, you can view your server at http://localhost:5000/
.
% ./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/
.
Check out the Getting Started page.