Linux下監(jiān)視NVIDIA的GPU使用情況詳解

在使用TensorFlow跑深度學(xué)習(xí)的時(shí)候，經(jīng)常出現(xiàn)顯存不足的情況，所以我們希望能夠隨時(shí)查看GPU時(shí)使用率。如果你是Nvidia的GPU，那么在命令行下，只需要一行命令就可以實(shí)現(xiàn)。

1. 顯示當(dāng)前GPU使用情況

Nvidia自帶了一個(gè)nvidia-smi的命令行工具，會(huì)顯示顯存使用情況：

$ nvidia-smi

輸出：

2. 周期性輸出GPU使用情況

但是有時(shí)我們希望不僅知道那一固定時(shí)刻的GPU使用情況，我們希望一直掌握其動(dòng)向，此時(shí)我們就希望周期性地輸出，比如每 10s 就更新顯示。 這時(shí)候就需要用到 watch命令，來周期性地執(zhí)行nvidia-smi命令了。

了解一下watch的功能：

$ whatis watch
 watch(1)    - execute a program periodically, showing output fullscreen

作用：周期性執(zhí)行某一命令，并將輸出顯示。

watch的基本用法是：

$ watch [options] command

最常用的參數(shù)是 -n， 后面指定是每多少秒來執(zhí)行一次命令。

監(jiān)視顯存：我們?cè)O(shè)置為每 10s 顯示一次顯存的情況：

$ watch -n 10 nvidia-smi

顯示如下：

這樣，只要開著這個(gè)命令行窗口，就可以每十秒刷新一次，是不是很方便呢？

如果我們希望來周期性地執(zhí)行其他命令行操作，那么就可以簡單地更換后面的nvidia-smi即可，So Cool !

以上就是本文的全部內(nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。