wookayin/gpustat

解決する課題

gpustat は、NVIDIA GPUの状態を監視するために nvidia-smi よりも簡潔で読みやすい代替手段を提供します。温度、利用率、メモリ使用量などのGPUメトリクスを簡略化して表示するため、どのプロセスやユーザーがGPUリソースを消費しているかを簡単に追跡できます。

仕組み

NVIDIAの公式Pythonバインディングである nvidia-ml-py (NVML) のラッパーとして機能します。コマンドラインツールとして実行することも、PythonライブラリとしてインポートしてプログラムからGPU統計を照会することも可能です。

対象ユーザー

NVIDIA GPUを使用する開発者や研究者で、ハードウェアの利用状況やプロセスの所有権をリアルタイムで迅速かつ明確に監視する必要がある方。

主な特徴

• 包括的なメトリクス: GPUインデックス、名前、温度、利用率、メモリ使用量を表示します。
• プロセスの追跡: プロセス所有者のユーザー名、プロセス名、PIDを表示できます。
• 柔軟な出力: カラーターミナル出力、自動化のためのJSON形式、定期更新のためのウォッチモードをサポートしています。
• Python API: 他のPythonプロジェクトに統合し、コードを通じてGPU統計を照会できます。