空闲显卡/ GPU 检测工具/程序
发布时间
阅读量：
阅读量
深度学习相关的训练过程通常需要消耗大量的时间，有时训练会在深夜完成，此时人员无法在计算机旁及时获知任务是否完成，导致GPU资源处于空闲状态，无法立即开展后续的训练工作。
为提升工作效率，可编写一段程序用于实时监控GPU的使用状态。
# author: Anthony
import os
import sys
import time
cmd = 'python main.py'
def gpu_info(gpu_index):
gpu_status = os.popen('nvidia-smi | grep %').read().split('\n')[gpu_index].split('|')
power = int(gpu_status[1].split()[-3][:-1])
memory = int(gpu_status[2].split('/')[0].strip()[:-3])
return power, memory
全部评论 (0)
还没有任何评论哟~