test_torch.py

import torch

print("✅ PyTorch is working!")
print("Version:", torch.__version__)
print("CUDA available:", torch.cuda.is_available())