深度学习的任务往往需要花费很多训练时间,有的时候训练可能在深夜结束,人不在电脑前无法第一时间知道任务结束情况,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 =