查看python是否安装了cuda

在Python中检查是否安装了CUDA（Compute Unified Device Architecture），通常需要通过第三方库`torch`或`cupy`来进行，因为它们依赖于CUDA。以下是两个简单的步骤：

1. **使用PyTorch**:
- 首先，你需要安装`torch`库（如果还没有安装）。打开命令行，输入：

     pip install torch torchvision

- 安装完成后，你可以导入`torch`并尝试运行下面的代码片段来检测CUDA支持：

     import torch
     if torch.cuda.is_available():
         print("CUDA is available.")
     else:
         print("CUDA is not available.")

2. **使用CuPy**:
- 如果你更喜欢CuPy（针对GPU加速的NumPy替代品），可以这样检测：

     try:
         import cupy
         print("CuPy found with CUDA support.")
     except ImportError:
         print("CuPy or CUDA not found.")

如果你的系统上没有安装CUDA，上述代码会显示“CUDA is not available.”或“CuPy or CUDA not found.”。