コマンドラインから持っているGPUカードの数を取得する方法は？

Question

CUDAツールを作成しようとしていますが、スケーラブルにする必要があります。システムがツールの引数として持つGPUの数を指定する必要があります。 GPUの数を直接取得するためのコマンドがあるかどうか教えてください。

Answer 1

Sudo lshw -C displayを使用してビデオカードをリストし、それぞれをグーグルで検索して、コアの数を確認できます。

Answer 2

このコマンドは、nvidia-smiがあると仮定して、GPUの数を直接取得します。

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

GPUの名前を1行に1つずつ出力し、行数をカウントします。