A certain engineer "COMPLEX"

開発メモ その11 GPUのメモリ使用量を可視化する

GeForce GTX 1080が欲しいです。

Caffeでどのくらいメモリを使っているのか?


ことの発端は、Caffeで学習開始時にCPUのメモリが足りなくて落ちてしまったこと。
個人的に、Caffeのログを見る限りそこまで大量のメモリを使っていない、と思っていたのですが、この際メモリの使用量を可視化してみることに。

GPUの状態は、C:\Program Files\NVIDIA Corporation\NVSMI\nvidia-smi.exeを実行するとわかります。


C:\WINDOWS\system32>"C:\Program Files\NVIDIA Corporation\NVSMI\nvidia-smi.exe"
Thu Aug 11 20:41:43 2016
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 368.39 Driver Version: 368.39 |
|-------------------------------+----------------------+----------------------+
| GPU Name TCC/WDDM | Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce GTX 750 Ti WDDM | 0000:01:00.0 Off | N/A |
| 53% 80C P0 27W / 38W | 1472MiB / 2048MiB | 90% Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| 0 624 C+G Insufficient Permissions N/A |
| 0 652 C+G Insufficient Permissions N/A |
| 0 680 C+G Insufficient Permissions N/A |
| 0 768 C+G Insufficient Permissions N/A |
| 0 1132 C+G Insufficient Permissions N/A |
| 0 2244 C+G Insufficient Permissions N/A |
| 0 5828 C+G ...indows.Cortana_cw5n1h2txyewy\SearchUI.exe N/A |
| 0 6376 C+G C:\Windows\explorer.exe N/A |
| 0 7108 C+G ...ost_cw5n1h2txyewy\ShellExperienceHost.exe N/A |
| 0 8308 C ...crosoft\caffe\Build\x64\Release\caffe.exe N/A |
+-----------------------------------------------------------------------------+

ちなみに引数で -lまたは --loop を付与すると、一定間隔で上記の出力を繰り返しで出力します。単位は秒。
なので、


C:\WINDOWS\system32>"C:\Program Files\NVIDIA Corporation\NVSMI\nvidia-smi.exe" -l 5

C:\WINDOWS\system32>"C:\Program Files\NVIDIA Corporation\NVSMI\nvidia-smi.exe" --loop=5

みたいな。

Conclusion


本当はGUIでグラフ表示的な感じで見たい。

コメントを残す

メールアドレスが公開されることはありません。

%d人のブロガーが「いいね」をつけました。