超算平台查看特定节点的显存使用情况

Posted 2022-12-21 CSU迦叶

srun --pty --nodelist=a40c01 nvidia-smi

其中 a40c01是计算节点的名称

如果你的conda环境安装了gpustat，上面的命令可以写成

 如果想看实时的(每0.1s刷新一次)

srun --pty --nodelist=a40c01 watch -n 0.1 nvidia-smi

易知gpustat可以换成nvidia-smi

最后，如果你不知道你的GPU集群有哪些节点

可以使用命令 

sinfo -N

 

但是不知道为什么有的节点就是不行，比如我这里的a30分区的每一个节点，还可以用指定分区的方式查看

srun --pty -p a30 watch -n 0.1 nvidia-smi

结果如下

 以上。