自动检测GPU是否空闲并执行程序

最新推荐文章于 2024-05-25 20:32:51 发布

Wanderer X

最新推荐文章于 2024-05-25 20:32:51 发布

阅读量346

点赞数 9

分类专栏：代码文章标签： python 开发语言

本文链接：https://blog.csdn.net/wandererxx/article/details/135571517

版权

代码专栏收录该内容

6 篇文章 0 订阅

订阅专栏

python代码

import os
import time
GPU_ID=2


while True:
    gpu_free_ls = os.popen('nvidia-smi --query-gpu=memory.free --format=csv').readlines()[1:]
    print(gpu_free_ls)
    gpu_free = int(gpu_free_ls[GPU_ID].strip().split()[0])
    print('gpu_free:', gpu_free)
    
    if gpu_free > 20000:
        os.system(f'''python generate.py --temp 0.9''')
        ...  
        break
        
    # 暂停5分钟再进行下一次扫描
    time.sleep(60*5)
    print('sleeping...')