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

aga*_*wal 4 command-line cuda gpu gpu-drivers

我正在尝试制作一个 CUDA 工具,并且必须使其具有可扩展性。我需要提供系统具有的 GPU 数量作为我工具的参数。请问有没有直接获取GPU数量的命令??

小智 9

nvidia-smi --list-gpus | wc -l
Run Code Online (Sandbox Code Playgroud)

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


小智 5

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

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

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

  • 更容易从记忆中输入:`nvidia-smi -L | 厕所-l` (4认同)

小智 3

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