欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

GPU排队脚本实现空闲触发python脚本实现示例

程序员文章站 2022-03-23 14:32:43
今天写了一个gpu排队脚本,事实上还是挺实用的。有的服务器是多用户使用,gpu的资源常常被占据着,很可能在夜间gpu空闲了,但来不及运行自己的脚本。如果没有和别人共享服务器的话,自己的多个程序想排队使...

今天写了一个gpu排队脚本,事实上还是挺实用的。有的服务器是多用户使用,gpu的资源常常被占据着,很可能在夜间gpu空闲了,但来不及运行自己的脚本。如果没有和别人共享服务器的话,自己的多个程序想排队使用gpu,也可以用这个脚本。环境非常简单,有python就行了:

python3.7

ubuntu16.04

先创建脚本:

vim narrow_setup.py

 cmd = 'python xxx.py'这句可以设置你要运行的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

就可以监听nvidia-smi中的信息,以伺机触发python脚本~

运行结果如下:

GPU排队脚本实现空闲触发python脚本实现示例

等待机会中... 如果gpu显存和功耗低于某个值时,就会触发python脚本。

以上就是gpu排队脚本实现一旦空闲就触发python脚本实现示例的详细内容,更多关于gpu排队脚本实现空闲触发python脚本的资料请关注其它相关文章!

相关标签: GPU 排队 脚本