如何检查我的系统上是否有可用的 NVIDIA GPU？

Question

有没有一种简单的方法可以仅使用标准库来检查 NVIDIA GPU 在我的系统上是否可用？我已经看到其他答案，他们建议使用 PyTorch 或 TensorFlow，但这不是我想要的。我想知道如何在 Windows 和 Linux 上执行此操作。谢谢！

Answer 1

安装 Nvidia 驱动程序后，该命令nvidia-smi会输出一个简洁的表格，其中提供有关 GPU、CUDA 和驱动程序设置的信息。

通过检查该命令是否存在，可以知道Nvidia GPU是否存在。

请注意，此代码仅在安装了 Nvidia GPU和适当的驱动程序时才有效。

这段代码应该可以在 Linux 和 Windows 上运行，它使用的唯一库是 subprocess，它是一个标准库。

import subprocess

try:
    subprocess.check_output('nvidia-smi')
    print('Nvidia GPU detected!')
except Exception: # this command not being found can raise quite a few different errors depending on the configuration
    print('No Nvidia GPU in system!')
    

Answer 2

以下代码显示 cuda 是否可用。cuda与gpu联系

print(torch.cuda.is_available())
print(torch.backends.cudnn.enabled)