NVIDIA DIGITS
The NVIDIA Deep Learning GPU Training System (DIGITS) puts the power of deep learning into the hands of engineers and data scientists. DIGITS can be used to rapidly train the highly accurate deep neural network (DNNs) for image classification, segmentation and object detection tasks.
DIGITS simplifies common deep learning tasks such as managing data, designing and training neural networks on multi-GPU systems, monitoring performance in real time with advanced visualizations, and selecting the best performing model from the results browser for deployment. DIGITS is completely interactive so that data scientists can focus on designing and training networks rather than programming and debugging.
DIGITS 4 is available as a free download to the members of the NVIDIA Developer Program. If you are not already a member, clicking “Download” will ask you join the program.
DIGITS 4 is available as a Amazon Machine Image (AMI) for on-demand usage. Get started instantly by clicking the “DIGITS AMI” button below. Learn more about GPU-accelerated cloud images from NVIDIA.
What’s New in DIGITS 5
- Train Image Segmentation neural networks to partition images into regions such as tumors from healthy tissue, and buildings and cars from free space
- Decrease training time and improve model accuracy with pre-trained models from the DIGITS Model Store
DIGITS 5 will be available in January, as a free download for members of the NVIDIA Developer Program. Sign up for the program to be notified when it is ready for download.
Learn more about DIGITS 5 in the following blog posts:
Image segmentation neural network trained with DIGITS 5 to partition epithelium regions that contribute to identification of tumor
Key Features:
- Design, train and visualize deep neural networks for image classification, segmentation and object detection
- Download pre-trained models from the DIGITS Model Store
- Perform hyperparameter sweep of learning rate and batch size for improved model accuracy
- Schedule, monitor, and manage neural network training jobs, and analyze accuracy and loss in real time
- Import a wide variety of image formats and sources
- Scale training jobs across multiple GPUs automatically
DIGITS is an open source project. Customize and extend DIGITS to suit your applications and share your experience using DIGITS on the DIGITS user group.
Import data for image classification and object detection neural networks
Choose predefined LeNet, AlexNet, GoogLeNet or provide custom networks
Visualize deep neural network architectures
Import data for image classification and object detection neural networks
Choose predefined LeNet, AlexNet, GoogLeNet or provide custom networks
Visualization of inference results
References:
Learn more about GPU-accelerated machine learning and deep learning technologies in these blog posts:
- Image Segmentation using DIGITS 5
- Deep Learning for Object Detection with DIGITS
- DetectNet: Deep Neural Network for Object Detection in DIGITS
- Easy Multi-GPU Deep Learning with DIGITS 2
- Introduction to Deep Learning with DIGITS webinar recording
- DIGITS: Deep Learning GPU Training System
- Accelerate Machine Learning with the cuDNN Deep Neural Network Library
- Deep Learning for Computer Vision with Caffe and cuDNN