python检测显卡cuda是否可用

代码示例

import torch
if __name__ == '__main__':
    print(torch.__version__)  # 查看 PyTorch 版本
    print(torch.cuda.is_available())  # 是否支持 CUDA
    print(torch.cuda.device_count())  # 查看 CUDA的数量
    print(torch.version.cuda)  # 查看 CUDA 版本（如果安装的是 GPU 版本）

运行结果，2.5.1+cpu表示是安装的cpu版本的torch

2.5.1+cpu
False
0
None

False表示不可以，0没有没有cuda设备

torch.cuda.is_available(): False
torch.cuda.device_count(): 0

如果需要重新安装 CUDA 版本的 PyTorch

卸载当前 PyTorch：

pip uninstall torch torchvision torchaudio

根据您的 CUDA 版本安装对应的 PyTorch（需与显卡驱动兼容）：

例如，PyTorch 官方下载页面 如果您的 CUDA 版本是 11.8，使用以下命令：

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118