背景

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

腳本

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

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。