GPU排隊腳本實現(xiàn)空閑觸發(fā)python腳本實現(xiàn)示例

更新時間：2021年11月01日 16:33:15 作者：木盞

有的服務器是多用戶使用，GPU的資源常常被占據(jù)著，很可能在夜間GPU空閑了，但來不及運行自己的腳本。如果沒有和別人共享服務器的話，自己的多個程序想排隊使用GPU，也可以用這個腳本

今天寫了一個GPU排隊腳本，事實上還是挺實用的。有的服務器是多用戶使用，GPU的資源常常被占據(jù)著，很可能在夜間GPU空閑了，但來不及運行自己的腳本。如果沒有和別人共享服務器的話，自己的多個程序想排隊使用GPU，也可以用這個腳本。環(huán)境非常簡單，有python就行了：

python3.7

ubuntu16.04

先創(chuàng)建腳本：

vim narrow_setup.py

cmd = 'python xxx.py'這句可以設(shè)置你要運行的python腳本

# author: muzhan
# contact: levio.pku@gmail.com
import os
import sys
import time 
cmd = 'python ~/hehe.py' 
def gpu_info():
    gpu_status = os.popen('nvidia-smi | grep %').read().split('|')
    gpu_memory = int(gpu_status[2].split('/')[0].split('M')[0].strip())
    gpu_power = int(gpu_status[1].split('   ')[-1].split('/')[0].split('W')[0].strip())
    return gpu_power, gpu_memory 
def narrow_setup(interval=2):
    gpu_power, gpu_memory = gpu_info()
    i = 0
    while gpu_memory > 1000 or gpu_power > 20:  # set waiting condition
        gpu_power, gpu_memory = gpu_info()
        i = i % 5
        symbol = 'monitoring: ' + '>' * i + ' ' * (10 - i - 1) + '|'
        gpu_power_str = 'gpu power:%d W |' % gpu_power
        gpu_memory_str = 'gpu memory:%d MiB |' % gpu_memory
        sys.stdout.write('\r' + gpu_memory_str + ' ' + gpu_power_str + ' ' + symbol)
        sys.stdout.flush()
        time.sleep(interval)
        i += 1
    print('\n' + cmd)
    os.system(cmd) 
if __name__ == '__main__':
    narrow_setup()

直接運行腳本：

python narrow_setup.py

就可以監(jiān)聽nvidia-smi中的信息，以伺機觸發(fā)python腳本~

運行結(jié)果如下：

等待機會中... 如果gpu顯存和功耗低于某個值時，就會觸發(fā)python腳本。

以上就是GPU排隊腳本實現(xiàn)一旦空閑就觸發(fā)python腳本實現(xiàn)示例的詳細內(nèi)容，更多關(guān)于GPU排隊腳本實現(xiàn)空閑觸發(fā)python腳本的資料請關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章:

相關(guān)文章

pytorch?collate_fn的基礎(chǔ)與應用教程
這篇文章主要給大家介紹了關(guān)于pytorch?collate_fn基礎(chǔ)與應用的相關(guān)資料,文中通過實例代碼介紹的非常詳細,對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下
2022-02-02
pytorch + visdom 處理簡單分類問題的示例
這篇文章主要介紹了pytorch + visdom 處理簡單分類問題的示例，小編覺得挺不錯的，現(xiàn)在分享給大家，也給大家做個參考。一起跟隨小編過來看看吧
2018-06-06
python網(wǎng)絡爬蟲精解之pyquery的使用說明
PyQuery是一個類似于jQuery的解析網(wǎng)頁工具，使用lxml操作xml和html文檔，它的語法和jQuery很像。和XPATH，Beautiful Soup比起來，PyQuery更加靈活，提供增加節(jié)點的class信息，移除某個節(jié)點，提取文本信息等功能
2021-09-09
Python3中內(nèi)置類型bytes和str用法及byte和string之間各種編碼轉(zhuǎn)換問題
這篇文章主要介紹了Python3中內(nèi)置類型bytes和str用法及byte和string之間各種編碼轉(zhuǎn)換問題,非常不錯，具有一定的參考借鑒價值，需要的朋友可以參考下
2018-09-09
pandas實現(xiàn)導出數(shù)據(jù)的四種方式
這篇文章主要介紹了pandas實現(xiàn)導出數(shù)據(jù)的四種方式，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2020-12-12
Numpy 數(shù)組操作之元素添加、刪除和修改的實現(xiàn)
本文主要介紹了Numpy 數(shù)組操作之元素添加、刪除和修改的實現(xiàn)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2023-03-03
Python實現(xiàn)多線程抓取妹子圖
本文給大家匯總了3款由Python制作的多線程批量抓取美圖的代碼，主要是將獲取圖片鏈接任務和下載圖片任務用線程分開來處理了，而且這次的爬蟲不僅僅可以爬第一頁的圖片鏈接的，有類似需求的小伙伴可以參考下。
2015-08-08
Pandas 內(nèi)置的十種畫圖方法
這篇文章主要介紹了Pandas 內(nèi)置的十種畫圖方法，Pandas是非常常見的數(shù)據(jù)分析工具，我們一般都會處理好處理數(shù)據(jù)然后使用searbon或matplotlib來進行繪制
2022-09-09
python sort、sort_index方法代碼實例
這篇文章主要介紹了python sort、sort_index方法，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2019-03-03
python中numpy.empty()函數(shù)實例講解
在本篇文章里小編給大家分享的是一篇關(guān)于python中numpy.empty()函數(shù)實例講解內(nèi)容，對此有興趣的朋友們可以學習下。
2021-02-02