方法一:使用命令行
nvidia-smi
是 Nvidia 顯卡命令行管理套件,基於 NVML 庫,旨在管理和監控 Nvidia GPU 設備。
這是 nvidia-smi 命令的輸出,其中最重要的兩個指標:
顯存占用
GPU 利用率
顯存占用和 GPU 利用率是兩個不一樣的東西,顯卡是由 GPU 計算單元和顯存等組成的,顯存和 GPU 的關系有點類似於內存和 CPU 的關系。
方法二:
找到一個好用的小工具:gpustat,
使用命令行
pip install gpustat
即可安裝,
gpustat 基於nvidia-smi,可以提供更美觀簡潔的展示,
使用下面 watch 命令,可以動態實時監控 GPU 的使用情況。
watch --color -n1 gpustat -cpu
目前沒有跑程序,所以顯存是沒有使用的。
關注公眾號:Python爬蟲數據分析挖掘,學習更多python知識