Repository Public Research preview
1#!/usr/bin/env bash2# One-shot setup of a Lambda Cloud GPU instance (Ubuntu, CUDA drivers preinstalled) for WP-AI-0002 training.3# Run ON the instance: bash lambda_setup.sh (repo must already be rsynced to ~/wpai0002, see lambda_sync.sh)4set -euo pipefail5cd ~/wpai00026nvidia-smi --query-gpu=name,memory.total,driver_version --format=csv,noheader || true7if ! command -v uv >/dev/null; then curl -LsSf https://astral.sh/uv/install.sh | sh >/dev/null 2>&1; fi8export PATH="$HOME/.local/bin:$PATH"9uv venv --python 3.12 .venv >/dev/null 2>&1 || true10uv pip install --python .venv/bin/python torch --index-url https://download.pytorch.org/whl/cu128 >/dev/null 2>&1 || uv pip install --python .venv/bin/python torch >/dev/null 2>&111uv pip install --python .venv/bin/python sentencepiece safetensors rapidfuzz sacrebleu rouge-score datasketch langdetect lxml xxhash pydantic numpy scipy pyarrow >/dev/null 2>&112.venv/bin/python -c "import torch; print('torch', torch.__version__, 'cuda', torch.cuda.is_available(), torch.cuda.get_device_name(0) if torch.cuda.is_available() else '-')"13.venv/bin/python -m pytest tests/unit -q -p no:warnings -k "models or losses or eval_audits" 2>&1 | tail -1 || true14echo "LAMBDA SETUP DONE"15