| #!/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") | |