diff --git a/param_decomp_lab/experiments/lm/launch.py b/param_decomp_lab/experiments/lm/launch.py
index ea9d285cd..41c3af383 100644
--- a/param_decomp_lab/experiments/lm/launch.py
+++ b/param_decomp_lab/experiments/lm/launch.py
@@ -100,7 +100,7 @@ def _node_workspace_setup(run_id: str, snapshot_ref: str, source_repo: Path, run
 git checkout --quiet FETCH_HEAD
 cp "{run_dir}/.env" .env
 unset VIRTUAL_ENV
-DRIVER_MAJOR=$(nvidia-smi --query-gpu=driver_version --format=csv,noheader | head -n 1 | cut -d. -f1)
+DRIVER_MAJOR=$(nvidia-smi --id=0 --query-gpu=driver_version --format=csv,noheader | cut -d. -f1)
 if [ "$DRIVER_MAJOR" -ge 580 ]; then CUDA_EXTRA=cuda13; else CUDA_EXTRA=cuda; fi
 echo "cuda extra: $CUDA_EXTRA (driver major $DRIVER_MAJOR)"
 uv sync --all-packages --no-dev --extra "$CUDA_EXTRA" --link-mode copy -q