How we keep GPUs reliable across Databricks AI

· Source: Databricks · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Cloud Computing & IT Infrastructure · Depth: Advanced, medium

Summary

Databricks AI details its approach to maintaining GPU reliability for massive-scale training workloads, addressing continuous failures across hardware, fabric, and software. The article identifies three primary failure categories: crashed jobs (often due to "NCCL watchdog timeout" messages), silent slowdowns (from degraded hardware like thermal throttling or interconnect issues, indicated by DCGM throttle reasons like "HW_SLOWDOWN" or "HW_THERMAL_SLOWDOWN"), and numerical corruption (memory faults or propagating errors). Given a conservative 1% annualized GPU failure rate, a 256-GPU job running for 30 days has a 19% chance of failure, rising to 57% for 1,024 GPUs, making failures expected. Databricks' solution involves stress testing with diverse, cutting-edge workloads and a multi-stage "gpu-monitor" health check system that operates across the node lifecycle.

Key takeaway

For MLOps Engineers building large-scale distributed GPU training infrastructure, you must anticipate and engineer for continuous hardware failures. Your reliability strategy should extend beyond basic crash detection to include identifying silent slowdowns and numerical corruption. Implement a multi-stage health check system, like Databricks' "gpu-monitor", to validate hardware pre-workload, monitor during execution, and verify inter-node fabric between jobs, ensuring robust and cost-effective model training.

Key insights

Distributed GPU training at scale necessitates proactive, multi-layered reliability engineering to counter expected hardware failures.

Principles

Method

Implement a multi-stage "gpu-monitor" service with active bootstrap checks (pre-workload), passive continuous checks (during workload), and periodic multi-node active checks (inter-workload) to cover the entire node lifecycle.

In practice

Topics

Best for: MLOps Engineer, Machine Learning Engineer, AI Architect

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Databricks.