St_Hakky’s blog

Data Science / Human Resources / Web Applicationについて書きます

GPUを使用する時によく使うコマンドとかまとめた

こんにちは。

GPUの種類の確認

lspci | grep -i nvidia

nvidia driverのバージョン確認コマンド

cat /proc/driver/nvidia/version

nvidia-smiコマンド

オプションはたくさんあります。よく使うコマンドとか、探している時に便利そうだなと思ったものを以下に列挙します。

オプション オプションで出力できる内容
オプションなし サマリ
-L, --list-gpus GPUを列挙
-i, --id= GPUを指定
-f, --filename= ファイルに出力
-l, --loop= 指定されたインターバル(デフォルト5秒)で表示を続ける。Ctrl+Cでとめる。
-q, --query GPU情報表示
-d, --display= 特定の項目だけ出力 カンマでつなげる。MEMORY,UTILIZATION, ECC, TEMPERATURE, POWER, CLOCK,COMPUTE, PIDS,PERFORMANCE, SUPPORTED_CLOCKS, PAGE_RETIREMENT, ACCOUNTING.

また、以下のコマンドとかはすごく便利。

# 10秒おきにGPUのサマリを表示
$ nvidia-smi -l 10

# watchで10病おきにCPUのサマリを表示
$ watch -n10 "nvidia-smi"

# GPUを使ってるプロセスを知る
$ nvidia-smi -q -d PIDS