首先，先检查系统的CUDA，然后再检查Python虚拟环境的CUDA，

在 Python 环境中检查 CUDA 的可用性，最核心和通用的方法是使用 PyTorch 库。这里为你提供了从基础到全面的验证步骤和代码。
检查CUDA版本主要有以下几种命令，最常用和直接的是前两种：

方法一：检查CUDA编译器版本（最直接）

这个方法通过查询NVIDIA CUDA编译器 nvcc 的版本来确认。

nvcc --version

输出示例：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Mon_Apr__3_17:16:06_PDT_2023
Cuda compilation tools, release 12.1, V12.1.105

关键信息：最后一行中的 release 12.1 表示系统安装的CUDA Toolkit版本是12.1。

方法二：检查NVIDIA驱动支持的CUDA版本

这个方法查看的是显卡驱动所能支持的最高CUDA版本，不一定是你实际安装的CUDA Toolkit版本。

nvidia-smi

输出示例：

+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.86.10              Driver Version: 535.86.10    CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+
...

关键信息：右上角 CUDA Version: 12.2 表示当前驱动支持的最高CUDA版本为12.2。

重要说明：nvidia-smi 显示的CUDA版本代表驱动兼容的版本。你实际通过 nvcc --version 安装的CUDA Toolkit版本应不高于此版本。两者可以不同，例如驱动支持12.2，但你完全可以安装并使用12.1的Toolkit。

其他辅助方法

如果上述命令不可用，可以尝试以下方法作为备选：

方法	命令	说明
查看CUDA运行时版本	cat /usr/local/cuda/version.txt	直接读取CUDA的版本文件（如果路径存在）。
通过环境变量查看	echo $CUDA_HOME 或 echo $PATH	检查CUDA的安装路径，然后通过路径名称推断版本。

总结与操作建议

1. 首选 nvcc --version：这是确认实际安装的CUDA Toolkit版本最准确的方法。
2. 再用 nvidia-smi 核对：确保你的CUDA Toolkit版本没有超过驱动所支持的最高版本。
3. 版本不匹配？ 如果Toolkit版本高于驱动支持版本，通常需要升级显卡驱动。

你可以先运行 nvcc --version 检查一下。如果命令未找到或结果不符合预期，请告诉我你的操作系统，我可以帮你排查问题所在。

🔍 基础核心检查

运行以下代码是最直接的验证方式：

import torch
# 1. 检查CUDA是否可用（返回True/False）
print('CUDA available:', torch.cuda.is_available())
# 2. 获取当前CUDA版本（PyTorch编译时所基于的版本）
print('CUDA version:', torch.version.cuda)
# 3. 获取GPU设备数量
print('GPU count:', torch.cuda.device_count())
# 4. 获取当前GPU设备名称
if torch.cuda.is_available():
    print('Current device:', torch.cuda.current_device())
    print('Device name:', torch.cuda.get_device_name(0))

📊 检查结果解读与常见状态

根据你的检查结果，可以参考下表快速定位问题：

检查结果	含义	可能的原因与下一步
torch.cuda.is_available() == True	✅ 环境正常。PyTorch已安装支持CUDA的版本，且系统驱动正常。	无需操作，可进行后续开发。
torch.cuda.is_available() == False	❌ CUDA不可用。	情况A：安装的是CPU版本PyTorch。 情况B：系统NVIDIA驱动未安装或版本不匹配。 情况C：PyTorch CUDA版本与系统CUDA运行时版本不匹配。
torch.version.cuda 显示版本号	PyTorch是GPU版本。	需与系统CUDA版本（见下文“系统检查”）大致兼容。
torch.version.cuda 显示 None	PyTorch是CPU版本。	需要重新安装支持CUDA的PyTorch。

🛠️ 深入与系统信息检查

如果CUDA不可用，可以从系统层面检查以定位问题。

1. 在Python中获取更多CUDA信息

import torch
print('PyTorch版本:', torch.__version__)
print('cuDNN版本:', torch.backends.cudnn.version())
# 检查计算能力（显卡兼容性）
if torch.cuda.is_available():
    for i in range(torch.cuda.device_count()):
        print(f'GPU {i} 计算能力:', torch.cuda.get_device_capability(i))

2. 在系统命令行中检查（更底层）
打开终端（CMD/PowerShell/Terminal）执行：

# 检查NVIDIA驱动和系统CUDA运行时版本
nvidia-smi

这个命令会显示系统安装的CUDA最高支持版本（右上角 CUDA Version 字样处），请确保 PyTorch 的 torch.version.cuda 版本号 ≤ 这里显示的系统CUDA版本。

如果 nvidia-smi 命令无效，说明NVIDIA显卡驱动可能没有正确安装，这是导致CUDA不可用的最常见原因。

💡 一个完整的验证脚本

你可以将以下代码保存为 check_cuda.py 并运行，它会输出一份完整的报告：

import torch
import sys

print('=' * 40)
print('PyTorch CUDA 可用性综合检测')
print('=' * 40)

print(f'Python 版本: {sys.version}')
print(f'PyTorch 版本: {torch.__version__}')
print(f'CUDA 编译版本: {torch.version.cuda}')

cuda_available = torch.cuda.is_available()
print(f'CUDA 是否可用: {cuda_available}')

if cuda_available:
    gpu_count = torch.cuda.device_count()
    print(f'可用GPU数量: {gpu_count}')
    for i in range(gpu_count):
        print(f'  GPU {i}: {torch.cuda.get_device_name(i)}')
        print(f'    内存总量: {torch.cuda.get_device_properties(i).total_memory / 1e9:.2f} GB')
        print(f'    计算能力: {torch.cuda.get_device_properties(i).major}.{torch.cuda.get_device_properties(i).minor}')
    current_device = torch.cuda.current_device()
    print(f'当前使用设备: GPU {current_device}')
else:
    print('-' * 40)
    print('CUDA不可用，可能的原因:')
    print('1. 安装的是PyTorch CPU版本。')
    print('2. NVIDIA显卡驱动未安装或版本太旧。')
    print('3. PyTorch CUDA版本与系统CUDA运行时版本不匹配。')

print('=' * 40)

📝 关键要点总结

1. 核心检查：始终以 torch.cuda.is_available() 返回 True 为最终目标。
2. 版本匹配：PyTorch的CUDA版本（torch.version.cuda）不应高于系统 nvidia-smi 显示的CUDA驱动版本。
3. 问题排查流程：
   • is_available() 为 False → 检查PyTorch是否为CPU版本（torch.version.cuda 为 None）。
   • 如果是CPU版本 → 使用正确的命令（如 pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121）重新安装GPU版本。
   • 如果是GPU版本但仍不可用 → 检查NVIDIA驱动（nvidia-smi 是否有效），并确保驱动版本支持PyTorch所需的CUDA版本。

你运行检查后结果如何？如果遇到问题，可以把 check_cuda.py 脚本的输出结果贴出来，我可以帮你具体分析。