lunzueta / DIGITS

Deep Learning GPU Training System

DIGITS Python3

My attempt at updating NVIDIA DIGITS for Python 3 compatibility.

Python 2.7 will not be maintained past 2020. Starting new projects in Python 2.7 is just asking for techinical debt you don't need.

Install instructions.

Tested on Ubuntu 16.04. But should work on in environment DIGITS and Python3 works.

Clone this project:

git clone https://github.com/jed-frey/DIGITS.git ~/DIGITS

To keep my OS's Python environment clean I prefer to work in Virtual Environments. Create a Python 3 virtual environment called digits_venv3 & activate it.

VENV=digits_venv3
virtualenv --python=$(which python3) ~/${VENV}
source ~/${VENV}/bin/activate

Setup DIGITS

cd ~/DIGITS
pip install -r requirements.txt
python setup.py build
python setup.py develop

Run Digits.

digits-devserver

or

digits-devserver --debug

Issues

It doesn't work just yet. No fancy mailing list, just use the issues list as it was intended.

https://github.com/jed-frey/DIGITS/issues

NVIDIA DIGITS ReadMe

DIGITS (the Deep Learning GPU Training System) is a webapp for training deep learning models. The currently supported frameworks are: Caffe, Torch, and Tensorflow.

Installation

Installation method	Supported platform[s]	Available versions	Instructions
Docker	Linux	DockerHub tags	nvidia-docker wiki
Source	Ubuntu 14.04, 16.04	GitHub tags	docs/BuildDigits.md

Usage

Once you have installed DIGITS, visit docs/GettingStarted.md for an introductory walkthrough.

Then, take a look at some of the other documentation at docs/ and examples/:

• Getting started with TensorFlow
• Getting started with Torch
• Fine-tune a pretrained model
• Train an autoencoder network
• Train a regression network
• Train a Siamese network
• Train a text classification network
• Train an object detection network
• Learn more about weight initialization
• Use Python layers in your Caffe networks
• Download a model and use it to classify an image outside of DIGITS
• Overview of the REST API

Get help

Installation issues

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

Usage questions

• First, check out the Getting Started page
• Then, ask questions on our user group

Bugs and feature requests

• 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 [e-mail] before your change can be accepted.