lqy845650069的博客

import osimport timeimport yagmailfrom loguru import loggerdef gpu_info(): gpu_status = os.popen('nvidia-smi | grep %').read().split('|') gpu_memory1 = int(gpu_status[2].split('/')[0].split('M')[0].strip()) gpu_memory2 = int(gpu_status[6]