import torch

# 查看 PyTorch 版本（关键！）
print(f"PyTorch 版本: {torch.__version__}")
# CPU 版本会显示: 2.6.0+cpu
# GPU 版本会显示: 2.6.0+cu118 / 2.6.0+cu121 / 2.6.0+cu124 等

# 检查 CUDA 是否可用
print(f"CUDA 可用: {torch.cuda.is_available()}")

# 如果有 CUDA，查看版本
if torch.cuda.is_available():
    print(f"CUDA 版本: {torch.version.cuda}")
    print(f"GPU 数量: {torch.cuda.device_count()}")
    print(f"GPU 名称: {torch.cuda.get_device_name(0)}")