检查自己的GPU相关信息和算力

不同显卡的算力不同，支持的pytorch版本CUDA版本都不同，如果使用不兼容的版本，可能会出现

what(): CUDA error: no kernel image is available for execution on the device

这样的报错。

可以使用下边的脚本进行检查

import torch
print("PyTorch 版本:", torch.__version__)
print("PyTorch CUDA 版本:", torch.version.cuda)

# 检查是否检测到 CUDA
print("CUDA 可用:", torch.cuda.is_available())
if torch.cuda.is_available():
    # 查看 GPU 型号
    gpu_model = torch.cuda.get_device_name(0)
    print("GPU 型号:", gpu_model)
    # 查看 GPU 算力（Compute Capability）
    gpu_arch = torch.cuda.get_device_capability(0)
    print("GPU 算力 (major.minor):", f"{gpu_arch[0]}.{gpu_arch[1]}")
else:
    print("未检测到 CUDA 设备（可能是 CPU 运行、GPU 不支持 CUDA 或驱动未安装）")