如何从命令行获取我拥有的 GPU 卡数量？

Question

我正在尝试制作一个 CUDA 工具，并且必须使其可扩展。我需要将系统拥有的 GPU 数量作为参数提供给我的工具。请告诉我是否有任何命令可以直接获取 GPU 数量？

Answer 1

假设您有，此命令直接获取 GPU 的数量nvidia-smi。

nvidia-smi --query-gpu=name --format=csv,noheader | wc -l

它打印 GPU 的名称，每行一个，然后计算行数。

Answer 2

nvidia-smi --list-gpus | wc -l

这将首先每行列出一个 GPU，然后计算输出的行数，即 GPU 的数量

Answer 3

您可以使用sudo lshw -C display列出您的视频卡，然后谷歌搜索每个视频卡以查看它有多少个核心。

Answer 4

如果您想在脚本中使用它，我建议像这样使用它：

nvidia-smi --query-gpu=name --format=csv,noheader | head -1

并验证号码：

GPU_num="$(nvidia-smi --query-gpu=name --format=csv,noheader | head -1)"
! [[ "$GPU_num" =~ ^[0-9]+$ ]] && echo "Not a number)"

最好数一下行数，因为行也可能是错误。