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 的名称,每行一个,然后计算行数。