GPU显存实时监测+显存空余时邮件提醒

三聚晴明

于 2021-04-03 20:10:34 发布

阅读量1.2k

点赞数 1

CC 4.0 BY-SA版权

文章标签： gpu 实时操作系统 pytorch

本文链接：https://blog.csdn.net/qq_30125323/article/details/115419929

人可以闲着，但机器不可以【狗头保命】。

查询当前可用GPU的可用空闲显存并在显存空余时发邮件提醒。还可以监测显存申请情况。提醒发现挖矿病毒或者其他恶意进程。

参考博文：Pytorch：实现查询当前可用GPU的可用空闲内存并按照内存从大到小顺序打印出来

保存如下代码到’gpu_status.py’

import numpy as np
import os
from email.mime.text import MIMEText
from email.mime.multipart import MIMEMultipart
import smtplib
import time

def send_mail(subject='No-reply', message='No-reply'):
    email_host = 'smtp.126.com'  # 服务器地址
    sender = '***@126.com'  # 发件人
    password = 'your_email_password_here'  # 密码，如果是授权码就填授权码
    receiver = '***@126.com'

最低0.47元/天解锁文章