test_env.py

import sys
import torch
import transformers

print("="*50)
print("VIRTUAL ENVIRONMENT TEST")
print("="*50)

# Check Python path
print(f"Python executable: {sys.executable}")
print(f"Virtual env active: {'venv' in sys.executable}")

# Check package versions
print(f"\nPackage Versions:")
print(f"PyTorch: {torch.__version__}")
print(f"Transformers: {transformers.__version__}")

# Check CUDA (GPU support)
print(f"\nGPU Available: {torch.cuda.is_available()}")
if torch.cuda.is_available():
    print(f"GPU Name: {torch.cuda.get_device_name(0)}")

print("\n✅ Virtual environment is working!")
print("="*50)

# device = torch.device("cpu")
# print(f"Using device: {device}")