Repolode
wookayin

wookayin/gpustat

NVIDIA、AMD、Huawei Ascend GPUの使用率、メモリ、温度、プロセスをコンパクトで読みやすく表示するコマンドラインツール。

代替できる製品:nvidia-smi

スター
4,387
フォーク
287
未解決の Issue
27
最終コミット
先月

これは何?

gpustatは、GPUの状態を照会・監視するための小さなPython製コマンドラインユーティリティです。デバイスごとに1行でGPU名、温度、使用率、メモリ使用量、実行中のプロセスを表示し、nvidia-smiよりコンパクトに状態を確認できます。

NVIDIA GPUはNVML、AMD GPUはROCm SMI、Huawei Ascend 910B NPUはnpu-smiを通じてサポートします。バックエンドは自動検出されますが、フラグで指定することもできます。ユーザー名、PID、コマンド全文、ファン速度、エンコーダー/デコーダー使用状況、消費電力などの列を追加でき、watchモードでは指定した間隔で出力を更新します。

CLIに加えて、gpustatは独自のコードからGPU統計情報を取得するPython APIを提供し、スクリプト向けにJSON形式で出力することもできます。

主な機能

  • 01デバイスごとに1行でGPU名、温度、使用率、メモリ使用量を表示します。
  • 02各GPUで実行中のプロセスについて、所有者、PID、メモリ使用量を一覧表示します。
  • 03NVIDIA(NVML)、AMD(ROCm SMI)、Huawei Ascend 910Bのバックエンドをサポートします。
  • 04指定した間隔で出力を更新するwatchモードを利用できます。
  • 05スクリプト向けのJSON出力とPython APIを提供します。

ユースケース

  • 共有ワークステーションやサーバーで、学習ジョブを開始する前に空いているGPUを確認する。
  • モデルの長時間学習中にGPUメモリと使用率を監視する。
  • 監視スクリプト内でGPUメトリクスをJSONとして収集する。
  • SSH経由でGPUの状態を手早く読みやすい形式で確認する。

クイックスタート

  1. 1

    PyPIからgpustatをインストールします

    pip install gpustat
  2. 2

    NVIDIA GPUを使用している場合は、NVIDIA公式のNVMLバインディング(pynvmlではありません)をインストールします

    pip install nvidia-ml-py
  3. 3

    gpustatを実行してGPUの状態を表示します

    gpustat

使う前に知っておきたいこと

  • NVIDIA GPUを使用するには、NVIDIAドライバー(gpustat 1.0以降では450.00以上)が必要です。
  • nvidia-ml-pyの代わりにpynvmlパッケージを使用しても動作しません。

収録されている特集

トピック

よくある質問

gpustat とは?
NVIDIA、AMD、Huawei Ascend GPUの使用率、メモリ、温度、プロセスをコンパクトで読みやすく表示するコマンドラインツール。 gpustatは、GPUの状態を照会・監視するための小さなPython製コマンドラインユーティリティです。デバイスごとに1行でGPU名、温度、使用率、メモリ使用量、実行中のプロセスを表示し、nvidia-smiよりコンパクトに状態を確認できます。
gpustat は無料で使えますか?ライセンスは?
gpustat は MIT ライセンスのオープンソースです。商用製品で使う前にライセンス条件を確認してください。
gpustat は今もメンテナンスされていますか?
半年以内に更新あり。gpustat の最終コミットは 2026年10月8日 です。
gpustat は何に使えますか?
共有ワークステーションやサーバーで、学習ジョブを開始する前に空いているGPUを確認する。 モデルの長時間学習中にGPUメモリと使用率を監視する。 監視スクリプト内でGPUメトリクスをJSONとして収集する。 SSH経由でGPUの状態を手早く読みやすい形式で確認する。
gpustat のインストール方法は?
PyPIからgpustatをインストールします: pip install gpustat NVIDIA GPUを使用している場合は、NVIDIA公式のNVMLバインディング(pynvmlではありません)をインストールします: pip install nvidia-ml-py gpustatを実行してGPUの状態を表示します: gpustat
gpustat はどこで入手できますか?
ソースコードとインストール手順は GitHub にあります:https://github.com/wookayin/gpustat

概要はプロジェクトの README をもとに AI が生成しています。GitHub の数値は毎日更新されます。

関連プロジェクト