快捷導(dǎo)航

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

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

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

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

python3.7

ubuntu16.04

先創(chuàng)建腳本：

vim narrow_setup.py

cmd = 'python xxx.py'這句可以設(shè)置你要運(yùn)行的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()

直接運(yùn)行腳本：

python narrow_setup.py

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

運(yùn)行結(jié)果如下：

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

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

您可能感興趣的文章:

相關(guān)文章

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