training_scripts/phase0_check.py

#!/usr/bin/env python3
"""Phase 0: Quick environment check"""

import torch

print("="*70)
print("PHASE 0: Environment Check")
print("="*70)

# GPU
if torch.cuda.is_available():
    print(f"\n✓ GPU: {torch.cuda.get_device_name(0)}")
    print(f"✓ Memory: {torch.cuda.get_device_properties(0).total_memory/1e9:.1f} GB")
else:
    print("\n❌ No GPU detected")
    exit(1)

# Packages
packages = ["transformers", "peft", "datasets", "trl", "vllm"]
missing = []

for pkg in packages:
    try:
        __import__(pkg)
        print(f"✓ {pkg}")
    except:
        missing.append(pkg)
        print(f"❌ {pkg}")

if missing:
    print(f"\n Install: pip install {' '.join(missing)}")
    exit(1)

print("\n✓ Ready to proceed")
print("\n➡️  Next: python phase1_data.py")