Skip to content
This repository was archived by the owner on Jul 17, 2026. It is now read-only.

fix(pd-lm): SIGPIPE-proof the driver-version detection in the node-setup script #990

Closed
claude-spd1 wants to merge 1 commit into from
Closed

fix(pd-lm): SIGPIPE-proof the driver-version detection in the node-setup script #990

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion param_decomp_lab/experiments/lm/launch.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down