python nvidia 显卡信息 格式数据

python nvidia 显卡信息 格式数据.

def get_gpu_memory():result = subprocess.check_output(['nvidia-smi', '--query-gpu=pci.bus_id,memory.used,memory.total,memory.free', '--format=csv'])# 返回 GPU 的显存使用情况，单位为 Minfo = []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

参考

• Python如何查看显存
• Useful nvidia-smi Queries