NVIDIA DGX-1: The Fastest Deep Learning System

One year ago today, NVIDIA announced the NVIDIA® DGX-1™, an integrated system for deep learning. DGX-1 (shown in Figure 1) features eight Tesla P100 GPU…

Mark Harris
10 min readadvanced
--
View Original

Overview

The article discusses the NVIDIA DGX-1, an integrated system designed for deep learning that features eight Tesla P100 GPU accelerators connected through NVLink. It highlights the system's architecture, performance capabilities, and software stack, emphasizing its unparalleled efficiency for deep learning training.

What You'll Learn

1

How to leverage NVLink for improved GPU communication in deep learning applications

2

Why InfiniBand is critical for multi-system scaling in deep learning environments

3

How to optimize deep learning training performance using the DGX-1 architecture

Prerequisites & Requirements

  • Understanding of deep learning concepts and GPU architecture
  • Familiarity with NVIDIA Docker and deep learning frameworks(optional)

Key Questions Answered

What is the architecture of the NVIDIA DGX-1 system?
The NVIDIA DGX-1 features eight Tesla P100 GPUs connected via NVLink in a hybrid cube-mesh topology, along with dual Intel Xeon CPUs and four 100 Gb InfiniBand network interface cards. This architecture maximizes throughput and interconnect bandwidth for deep learning tasks.
How does NVLink enhance deep learning performance compared to PCIe?
NVLink provides an aggregate bidirectional bandwidth of 160 GB/s per GPU, which is five times that of PCIe Gen3 x16 connections. This allows for better scaling and communication between GPUs, resulting in about 1.4x higher training performance when using eight GPUs compared to PCIe.
What role does InfiniBand play in the DGX-1 system?
InfiniBand in the DGX-1 system provides significant bandwidth for multi-system scaling, supporting 800 Gb/s total bandwidth with low latency. This enables efficient communication between multiple DGX-1 systems, essential for large-scale deep learning workloads.
What software stack is included with the DGX-1?
The DGX-1 software stack includes the NVIDIA CUDA Toolkit, NVIDIA Docker, the NVIDIA Deep Learning SDK, and optimized deep learning frameworks like Caffe and TensorFlow. This integrated software platform is designed to facilitate deep learning deployment with minimal setup.

Key Statistics & Figures

Aggregate bidirectional bandwidth per GPU
160 GB/s
This bandwidth is achieved through NVLink connections, significantly enhancing inter-GPU communication.
Performance improvement of DGX-1 over PCIe
1.4x
This improvement is observed when using eight GPUs for training deep learning models, demonstrating the efficiency of the NVLink architecture.
Total bandwidth provided by InfiniBand ports
800 Gb/s
This bandwidth is available for inter-system communication, crucial for multi-node deep learning setups.

Technologies & Tools

Hardware
Tesla P100
Used as the primary GPU accelerators for deep learning tasks in the DGX-1.
Interconnect
Nvlink
Facilitates high-speed communication between GPUs in the DGX-1 system.
Networking
Infiniband
Provides high bandwidth and low latency communication between multiple DGX-1 systems.
Software
Nvidia Docker
Enables easy deployment of deep learning frameworks and applications in containerized environments.

Key Actionable Insights

1
Utilize the NVLink architecture to enhance GPU communication and training efficiency in deep learning tasks.
By leveraging NVLink's high bandwidth capabilities, practitioners can achieve faster training times and better resource utilization when scaling deep learning models.
2
Implement InfiniBand for multi-system configurations to reduce latency and improve throughput in large-scale deep learning applications.
InfiniBand's high-speed communication capabilities are essential for managing data transfer between multiple DGX-1 systems, ensuring optimal performance in distributed training scenarios.
3
Take advantage of the pre-optimized software stack provided with DGX-1 to streamline deep learning project setups.
The availability of NVIDIA Docker containers with pre-built frameworks allows users to quickly deploy and test deep learning models without extensive configuration.

Common Pitfalls

1
Neglecting the importance of proper interconnect technology can lead to suboptimal performance in deep learning tasks.
Many users may default to PCIe without realizing that NVLink and InfiniBand offer significant advantages in terms of bandwidth and latency, which are critical for scaling deep learning applications.

Related Concepts

Deep Learning Frameworks
GPU Architecture
High-performance Computing
Distributed Systems