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

参考

相关推荐
逐步前行29 分钟前
C标准库--C99--布尔型<stdbool.h>
c语言·开发语言
程序员爱钓鱼30 分钟前
Python编程实战 · 基础入门篇 | 元组(tuple)
后端·python·ipython
QX_hao30 分钟前
【Go】--闭包
开发语言·golang
程序员爱钓鱼31 分钟前
Python编程实战 · 基础入门篇 | 列表(list)
后端·python·ipython
御承扬2 小时前
编程素养提升之EffectivePython(Builder篇)
python·设计模式·1024程序员节
chenchihwen3 小时前
AI代码开发宝库系列:FAISS向量数据库
数据库·人工智能·python·faiss·1024程序员节
林月明4 小时前
【VBA】自动设置excel目标列的左邻列格式
开发语言·excel·vba·格式
喜欢吃燃面4 小时前
数据结构算法题:list
开发语言·c++·学习·算法·1024程序员节
。TAT。4 小时前
C++ - 多态
开发语言·c++·学习·1024程序员节
AI视觉网奇4 小时前
json 可视化 2025 coco json
python·1024程序员节