Nvidia显卡数据获取

查看显卡列表

nvidia-smi -L

查看显卡显存

nvidia-smi --query-gpu=memory.total --format=csv,noheader,nounits

查看进程显存占用

nvidia-smi --query-compute-apps=pid,process_name,used_gpu_memory --format=csv,noheader,nounits