trainer/requirements_train_GPU.txt

# ═══════════════════════════════════════════════════════════
# VERIDEX v4.0 — Trainer Requirements (GPU: RTX 3050 CUDA 12.x)
# ═══════════════════════════════════════════════════════════

# PyTorch GPU — CUDA 12.1 (RTX 3050 compatible)
--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.3.0+cu121
torchvision==0.18.0+cu121

# Model architecture
timm==1.0.3

# Data processing
numpy==1.26.4
Pillow==10.3.0
tqdm==4.66.4
scikit-learn==1.5.0

# Visualization (optional)
matplotlib==3.9.0