# ==========================================
# CELL 2: IMPORTS & CUDA CHECK
# ==========================================
# Uncomment the line below and run ONCE if you need to install packages:
# !pip install unsloth transformers datasets trl accelerate bitsandbytes

import torch
import gc
import os

from unsloth import FastLanguageModel, standardize_sharegpt
from datasets import load_dataset
from transformers import TrainingArguments
from trl import SFTTrainer

print("=" * 60)
print("CUDA / GPU CHECK")
print("=" * 60)

if torch.cuda.is_available():
    print(f"GPU          : {torch.cuda.get_device_name(0)}")
    print(f"VRAM Total   : {torch.cuda.get_device_properties(0).total_memory / 1024**3:.1f} GB")
    print(f"VRAM Current : {torch.cuda.memory_allocated() / 1024**3:.1f} GB")
    print(f"BF16 Support : {torch.cuda.is_bf16_supported()}")
else:
    print("WARNING: CUDA not available. Training will be extremely slow on CPU.")

print("\nAll imports loaded successfully.")