spark-training-gotchas
Helps your agent predict and fix common ML task issues. Use before long tasks to avoid failures, slowdowns, or memory errors.
What it does
This tool helps find and fix common issues that can cause your AI training to fail or slow down on the DGX Spark system. It checks for ten known types of failures.
When to use it
- ✓Your training fails to start with strange errors.
- ✓Your training runs out of memory and crashes, even when it seems there's space.
- ✓Your training starts fast but then slows down.
Example of what to ask your AI
“My training on DGX Spark is crashing, what could be wrong?”
Want more? Open the sections below.
How to use
The easiest way: copy the message below and paste it into Claude, ChatGPT, Cursor or any AI. It reads the skill and follows the instructions.
Skill link
Ready-to-paste message
How it works
The tool analyzes your system and checks if any of the ten known problems are occurring. It tells you what the issue is and how to try to fix it.
Installation
Easiest way: copy the skill link and paste it into Claude, ChatGPT or Cursor asking it to follow the instructions.
In the terminal (Claude Code, Codex, Cursor), in your project folder:
npx skills add wshobson/agents --skill spark-training-gotchasOr download just the file:
curl -L https://raw.githubusercontent.com/wshobson/agents/main/plugins/dgx-spark-ops/skills/spark-training-gotchas/SKILL.md -o SKILL.mdTechnical content (original SKILL.md)
Original text by the author, straight from GitHub.
Spark Training Gotchas
DGX Spark's GB10 chip (Grace Blackwell, SM121, 128GB unified memory, aarch64) has ten recurring failure modes across launch, memory, thermals, bandwidth, and precision. Each is named G1–G10 so it can be checked by number — the numbering is load-bearing for tooling that runs these checks. Read this before a long run, not after hour six.
When to Use This Skill
- A training run fails to start, with an import error or a segfault that doesn't point at the real cause.
- A run OOMs while
nvidia-smistill shows headroom. - Throughput degrades partway through a run that started fine.
- Before any multi-hour or multi-epoch job on GB10.
- Wiring two Sparks together, before picking a parallelism strategy.
- Choosing between FP8 and NVFP4 for a Spark-hosted run.
Common Issues Quick Reference
| # | Symptom | Fix |
|---|---|---|
| G1 | undefined symbol / segfault | cu130 wheel or container |
| G2 | flash-attn wrong backend used | skip pip build; monkeypatch on NGC |
| G3 | OOM despite headroom | drop page cache |
| G4 | throughput drop / reboot | expect ~100W sustained cap |
| G5 | memory-bound step slow | budget 180–192 GB/s |
| G6 | cache evicted mid-run | one GPU server at a time |
| G7 | NVFP4 slower than FP8 | stay FP8 unless sm_121a |
| G8 | playbook fails outright | check upstream issues |
| G9 | env breaks after install | use a container |
| G10 | 2-Spark TP hangs | DDP/FSDP only, never TP |
The Ten Gotchas
G1: CUDA 12/13 ABI Mismatch
- SYMPTOM:
ImportError: undefined symbolnaming a CUDA function, or a segfault on the first.cuda()call. - CAUSE: most PyPI wheels link
libcudart.so.12; Spark ships CUDA 13. pip never checks CUDA ABI, so it surfaces only at import or first kernel launch. - CHECK:
references/gotcha-checks.mdG1 — the wheel's CUDA build tag. - FIX: reinstall from
download.pytorch.org/whl/cu130or use a matched container.
G2: flash-attn — Skip the pip Build, Watch Unsloth's Auto-Detect
- SYMPTOM:
pip install flash-attnstill fails/hangs. Unsloth may also silently train flash-attn over an explicitly requested SDPA. - CAUSE: no aarch64/sm_121 wheel for bare pip — but NGC
containers ship a working SM121 flash-attn, and Unsloth
auto-prefers it, dropping
attn_implementation="sdpa". - CHECK:
references/gotcha-checks.mdG2 — is flash-attn already present and working. - FIX: bare pip — skip flash-attn, use SDPA (unchanged). On
NGC — the only reliable override is the monkeypatch in
references/gotcha-checks.mdG2.
G3: UMA OOM Below 128GB
- SYMPTOM: OOM during model load/training while
nvidia-smistill reports free memory under the 128GB cap — or, on some setups,[N/A]outright instead of a number. - CAUSE: mmap and the CUDA allocator double-count pages during safetensors load; QLoRA can OOM earlier than bf16 since dequantization adds transient allocs.
- CHECK:
references/gotcha-checks.mdG3 — readfree -gand/proc/meminfo, notnvidia-smi. - FIX: drop the page cache with
sync; echo 3 > /proc/sys/vm/drop_caches— needs root, a between-run reset, not a mid-training step.
G4: Thermal Throttling
- SYMPTOM: throughput drops partway through a multi-hour run, or the box spontaneously reboots under sustained load.
- CAUSE: sustained power draw caps around 100W versus the 240W rated figure; long runs push into that ceiling and throttle or, sometimes, reboot.
- CHECK:
references/gotcha-checks.mdG4 — samplenvidia-smi --query-gpu=temperature.gpu,power.draw. - FIX: if power plateaus under 240W while temperature climbs, treat throttling as the cause; improve cooling or cap run length.
G5: Bandwidth Ceiling
- SYMPTOM: memory-bound workloads, decode-heavy RL loops especially, plateau well below expected throughput.
- CAUSE: 273 GB/s is a spec ceiling, not sustained; measured bandwidth runs 180–192 GB/s.
- CHECK:
references/gotcha-checks.mdG5 — observed step time vs. the measured range, not spec. - FIX: budget throughput from 180–192 GB/s; revise a plan built on the 273 GB/s figure.
G6: Global UMA Resource Contention
- SYMPTOM: a process's KV cache/weights get evicted mid-run silently, no OOM in its own logs.
- CAUSE: unified memory is one global pool; an uncapped or near-capacity process competes with anything else and can evict it. A small,
…
FAQ
What is DGX Spark?
It's a type of powerful computer designed for training artificial intelligence models quickly.
What are 'failure modes'?
These are the ways something can go wrong or fail.
Do I need to know how to code to use this?
No, the tool is designed to be used even without programming knowledge. It guides you with instructions.
Details
Original description by the author
Preflight and diagnose the ten known failure modes for ML training on NVIDIA DGX Spark. Use when a training run on DGX Spark fails to start, OOMs below the 128GB limit, slows down mid-run, or before any multi-hour training job on GB10.
repo: wshobson/agents
Popularity
- GitHub stars
- 41k
- Ranking position
- #4854
- License
- MIT
- Updated
- 10/15/2026
Comments
- …
More skills from this repository
screen-reader-testing
Has your agent test websites using screen readers to ensure they are accessible.
wshobson/agents
wcag-audit-patterns
Helps your agent audit websites for accessibility and fix issues according to WCAG guidelines.
wshobson/agents
multi-reviewer-patterns
Enables your agent to coordinate parallel code reviews, deduplicating findings and consolidating reports.
wshobson/agents
parallel-debugging
Allows the agent to debug complex issues with parallel investigations and root cause analysis.
wshobson/agents
Similar skills
bun-runtime
Has your agent use Bun to run, organize, and test code. It helps choose Bun over other tools and migrate existing projects.
affaan-m/ECC
database-migrations
Helps your agent make safe changes to databases. It ensures changes can be undone and minimizes downtime.
affaan-m/ECC
deployment-patterns
Sets up deployment and CI/CD workflows, containerization, and rollback strategies to ensure safe releases.
affaan-m/ECC