← ClaudeAtlas

gpu-custom-setuplisted

One-time onboarding flow for users who set ROCKIE_GPU_MODE=custom (i.e., they have their own GPU setup — own AWS account, on-prem cluster, SSH tunnel to a workstation, university HPC, custom orchestration — instead of using Rockie's deidentified GPU router). Trigger this when (1) the user mentions GPU/training/provisioning, (2) `echo $ROCKIE_GPU_MODE` returns `custom`, AND (3) `.claude/gpu-custom.md` doesn't exist or is empty. Walks the user through their auth/provision/connect/monitor/terminate flow and saves it to .claude/gpu-custom.md so future agent sessions reuse the saved flow without re-asking. Run this AT MOST ONCE per project.
Rockielab/rockie-claude · ★ 20 · AI & Automation · score 76
Install: claude install-skill Rockielab/rockie-claude
# /gpu-custom-setup — onboard a user with their own GPU setup Rockie's default GPU layer is the deidentified GPU router (`gpu.py`). But some users have their own setup they prefer to drive themselves — their own AWS account, on-prem cluster, university HPC quota, custom SSH tunnel, etc. They opt out by setting `ROCKIE_GPU_MODE=custom`, and the agent reaches for THIS skill on first GPU need to learn how THEY provision compute. ## Pre-flight checks Run these BEFORE invoking the skill. If any fail, do not proceed. ```bash # 1. Are we in custom mode? test "$(echo "${ROCKIE_GPU_MODE:-router}")" = "custom" || exit 0 # 2. Has setup already been done? test -s .claude/gpu-custom.md && { echo "[gpu-custom-setup] .claude/gpu-custom.md already populated — skipping" exit 0 } ``` If both checks pass, continue. ## What to do This is a Q&A skill — the agent prompts the user, the user pastes commands or describes their flow, the agent structures the answers into `.claude/gpu-custom.md` and saves. Sections to elicit, **in this order**: ### 1. Authentication Ask: "How do you authenticate to your GPU provider? Paste any commands or env vars you set up. (e.g. `aws configure`, `gcloud auth login`, `ssh-add ~/.ssh/key`, custom token in env, etc.)" Probe: if they say "AWS", ask whether they use IAM role, access keys, or SSO. If "on-prem", ask whether SSH key auth or password. ### 2. Provision (start a GPU) Ask: "How do you spin up a GPU? Paste the command, script, or describe the s