import torch
print("CUDA 版本：", torch.version.cuda)  # 应输出 12.8
print("cuDNN 版本：", torch.backends.cudnn.version())  # 应输出 9200+
print("GPU 可用：", torch.cuda.is_available())  # 应输出 True