import torch

print("PyTorch version:", torch.__version__)
print("CUDA available:", torch.cuda.is_available())
print("CUDA compiled version:", torch.version.cuda)
print("GPU count:", torch.cuda.device_count())