refactor: add logging for CUDA and PyTorch versions

generate_summaries_uv.py

@@ -36,6 +36,19 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+# log cuda version
+import torch
+if torch.cuda.is_available():
+    logger.info(f"CUDA version: {torch.version.cuda}")
+    logger.info(f"CUDA device: {torch.cuda.get_device_name(0)}")
+    logger.info(f"CUDA capability: {torch.cuda.get_device_capability(0)}")
+else:
+    logger.info("CUDA is not available")
+
+# log torch version
+logger.info(f"PyTorch version: {torch.__version__}")
+logger.info(f"vLLM version: {vllm.__version__}")
+
 
 def format_prompt(content: str, card_type: str, tokenizer) -> str:
     """Format content as a prompt for the model."""