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

参考

相关推荐
西柚小萌新9 分钟前
【深入浅出PyTorch】--上采样+下采样
人工智能·pytorch·python
rit843249937 分钟前
基于MATLAB的模糊图像复原
开发语言·matlab
fie888942 分钟前
基于MATLAB的声呐图像特征提取与显示
开发语言·人工智能
_extraordinary_2 小时前
Java SpringMVC(二) --- 响应,综合性练习
java·开发语言
shut up2 小时前
LangChain - 如何使用阿里云百炼平台的Qwen-plus模型构建一个桌面文件查询AI助手 - 超详细
人工智能·python·langchain·智能体
宝贝儿好3 小时前
【python】第五章:python-GUI编程
python·pyqt
@。1243 小时前
对于灰度发布(金丝雀发布)的了解
开发语言·前端
闲人编程3 小时前
从多个数据源(CSV, Excel, SQL)自动整合数据
python·mysql·数据分析·csv·存储·数据源·codecapsule
hsjkdhs4 小时前
C++之多层继承、多源继承、菱形继承
开发语言·c++·算法
B站_计算机毕业设计之家4 小时前
推荐系统实战:python新能源汽车智能推荐(两种协同过滤+Django 全栈项目 源码)计算机专业✅
大数据·python·django·汽车·推荐系统·新能源·新能源汽车