背景

常常有這樣的情況，服務(wù)器上的GPU正被占著使用，我不可能有精力經(jīng)常來看看是不是其他用戶跑完了程序，也可能其他用戶跑完了但我不在電腦前，因此如果由一個(gè)自動(dòng)腳本幫我查看GPU是否空閑，一旦空閑就自動(dòng)運(yùn)行我想要跑的程序就好了，這就是我寫這個(gè)腳本的原因所在。本腳本不依賴任何第三方庫，如gpustat，只需要安裝正確的顯卡驅(qū)動(dòng)即可。

腳本

這個(gè)腳本的整體思路是非常簡(jiǎn)單的，就是過一段時(shí)間就自動(dòng)檢查指定顯卡的剩余顯存是否滿足我的需要，如滿足則運(yùn)行要跑的程序。這個(gè)腳本（run.sh）的內(nèi)容及解釋如下所示，該程序針對(duì)單卡程序，需要多卡的自行魔改即可。

#!/bin/bash while true do# 計(jì)算指定卡的剩余顯存大小LEFT=`nvidia-smi -a -i $1|grep Free

總結(jié)

以上是生活随笔為你收集整理的Linux闲时自动抢占GPU脚本的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。