タグ

ブックマーク / qiita.com/tomotaka_ito (1)

  • PythonからGPU使用率などの情報を取得する(nvidia-smiコマンド) - Qiita

    サーバのどのGPUが空いてるか調べたい→プログラムからGPU使用率とか取れないのかな〜と思ってちょっと調べたらいつも使っているnvidia-smiコマンドはオプション次第でマシンリーダブルな出力を出してくれそうなことがわかりました。 --query-gpuオプション --query-gpuオプションは欲しい情報を並べるとそれをとってきて--formatオプションで指定した出力スタイルに加工して出力してくれます。 たとえば以下のような情報を指定することができるようです。 index: 0からはじまるハードウェアのマシン上の番号, "0"とか"1"とか memory.free: 空きGPUメモリ量, "5123 MiB" とか memory.used: 使用GPUメモリ量, "3451 MiB" とか memory.total: GPUに搭載されているメモリ量, "8113 MiB" とか u

    PythonからGPU使用率などの情報を取得する(nvidia-smiコマンド) - Qiita
    aont
    aont 2023/01/06
  • 1