Linux闲时自动抢占GPU脚本

Posted 2021-06-14 周先森爱吃素

背景

常常有这样的情况，服务器上的GPU正被占着使用，我不可能有精力经常来看看是不是其他用户跑完了程序，也可能其他用户跑完了但我不在电脑前，因此如果由一个自动脚本帮我查看GPU是否空闲，一旦空闲就自动运行我想要跑的程序就好了，这就是我写这个脚本的原因所在。本脚本不依赖任何第三方库，如gpustat，只需要安装正确的显卡驱动即可。

脚本

这个脚本的整体思路是非常简单的，就是过一段时间就自动检查指定显卡的剩余显存是否满足我的需要，如满足则运行要跑的程序。这个脚本（run.sh）的内容及解释如下所示，该程序针对单卡程序，需要多卡的自行魔改即可。

#!/bin/bash
while true
do
  # 计算指定卡的剩余显存大小
	LEFT=`nvidia-smi -a -i $1|grep Free|