python nvidia 显卡信息 格式数据

python nvidia 显卡信息 格式数据.

python 复制代码
def get_gpu_memory():
    result = subprocess.check_output(['nvidia-smi', '--query-gpu=pci.bus_id,memory.used,memory.total,memory.free', '--format=csv'])
    # 返回 GPU 的显存使用情况,单位为 M
    info = []
    for t in csv.DictReader(result.decode().splitlines()):
        info.append({'id':t['pci.bus_id'],
         'used':int(t[' memory.used [MiB]'].split()[0]),
         'total':int(t[' memory.total [MiB]'].split()[0]),
         'free':int(t[' memory.free [MiB]'].split()[0])})
    return info

参考

相关推荐
databook6 小时前
Manim实现脉冲闪烁特效
后端·python·动效
程序设计实验室6 小时前
2025年了,在 Django 之外,Python Web 框架还能怎么选?
python
倔强青铜三8 小时前
苦练Python第46天:文件写入与上下文管理器
人工智能·python·面试
用户25191624271111 小时前
Python之语言特点
python
刘立军11 小时前
使用pyHugeGraph查询HugeGraph图数据
python·graphql
数据智能老司机15 小时前
精通 Python 设计模式——创建型设计模式
python·设计模式·架构
数据智能老司机16 小时前
精通 Python 设计模式——SOLID 原则
python·设计模式·架构
c8i17 小时前
django中的FBV 和 CBV
python·django
c8i17 小时前
python中的闭包和装饰器
python
这里有鱼汤20 小时前
小白必看:QMT里的miniQMT入门教程
后端·python