我使用tmux常常将一块屏幕的四分之一用于观察gpu利用率和显存剩余,但是如果我使用nvidia-smi就会显示不全,因为我有10块gpu。我想了想,直接使用nvidia-smi显示的信息很多是我不需要的,我只需要gpu-id号,显存剩余,显存总量,gpu利用率就这些,那么我们可以设置只显示这些:
nvidia-smi --query-gpu=index,utilization.gpu,memory.free,memory.total --format=csv,noheader,nounits
在这个命令中:
index
表示 GPU 的 ID。utilization.gpu
表示 GPU 的利用率。memory.free
表示显存剩余。memory.total
表示显存总量。
--format=csv,noheader,nounits
:
csv
:指定输出格式为逗号分隔的值。noheader
:指定不显示列标题。nounits
:指定不显示单位。
如果我们要每隔一秒显示:
watch -n 1 'nvidia-smi --query-gpu=index,utilization.gpu,memory.free,memory.total --format=csv,noheader,nounits'
标签:显存,伟达,--,smi,memory,gpu,csv From: https://www.cnblogs.com/chentiao/p/18033967