python - 如何确定 GPU 在我的操作系统上是否可用？

Question

现在人们正在谈论 GPU 计算。我如何知道我的计算机上是否有可用的 GPU？我想要一种 python 方式，或者使用 shell 命令来了解 GPU 是否可用。我使用的是 Mac 操作系统Catalina 版本 10.15(如果有的话)。

顺便说一句，我可以放心地假设当今大多数计算机都可以使用 GPU 吗？

Answer 1

我认为使用 PyTorch 库你可以确定这一点。通常，CUDA平台用于在 GPU 上进行计算。

因此，在 Python 中你必须执行以下操作:

import torch

torch.cuda.is_available()

无论 GPU/CUDA 是否可用，都会返回此信息。

然后您可以执行以下操作来选择设备(CPU 或 GPU)

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")