python中查看cuda版本

在Python中，如果你想要查看CUDA（Compute Unified Device Architecture）的版本，通常你需要使用第三方库`pynvml`或者`torch`（如果你打算进行深度学习相关的操作）。以下是两种方法：

1. 使用`pynvml`库：
安装`pynvml`后，你可以这样做：

   import pynvml
   pynvml.nvmlInit()  # 初始化NVML
   handle = pynvml.nvmlDeviceGetHandleByIndex(0)  # 获取主GPU设备句柄
   version_info = pynvml.nvmlSystemGetDriverVersion(handle)  # 获取驱动版本
   print("CUDA Version:", version_info)

请确保你已经安装了`pynvml`，如果没有，可以通过`pip install pynvml`来安装。

2. 使用`torch`库：
如果你的目的是检查PyTorch是否支持CUDA以及其版本，可以这样：

   import torch
   print("CUDA Version:", torch.cuda.get_device_properties(0).driver_version)

这会显示当前激活GPU的CUDA驱动版本。

注意，上述代码可能需要管理员权限才能运行，并且如果系统中没有CUDA或者`pynvml`库没有找到，可能会抛出异常。