获取指定进程所占用内存、指定GPU显存信息、所有GPU型号和驱动

#########################################################
# 获取指定进程所占用内存
#########################################################import os,psutilprocess = psutil.Process(25927)# 指定进程。当前进程用os.getpid()获得
memInfo = process.memory_info()
print ('内存占用:',memInfo.rss/1024/1024/1024,'G\n')#########################################################
# 获取指定GPU显存信息
#########################################################import pynvmlpynvml.nvmlInit()
handle = pynvml.nvmlDeviceGetHandleByIndex(3)# GPU id
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print("0:used: {} G".format(meminfo.used / 1024 / 1024/ 1024)) # 已用显存
print("total: {} G".format(meminfo.total/ 1024 / 1024/ 1024)) # 全部显存
print("free: {} G".format(meminfo.free/ 1024 / 1024/ 1024)) # 剩余显存

#########################################################
# 获取GPU型号和驱动版本
#########################################################
from pynvml import *
nvmlInit()print ("Driver Version:", str(nvmlSystemGetDriverVersion(),'utf-8'))# 显卡驱动版本deviceCount = nvmlDeviceGetCount()
for i in range(deviceCount):handle = nvmlDeviceGetHandleByIndex(i)print("Device", i, ":", str(nvmlDeviceGetName(handle), 'utf-8')) # 每一个显卡的型号nvmlShutdown()

获取指定进程所占用内存、指定GPU显存信息、所有GPU型号和驱动相关推荐

GPU显存满了, 但是GPU利用率很低
训练效率低?GPU利用率上不去?快来看看别人家的tricks吧- batch size太小,导致模型的并行化程度太低.还有你的cpu.内存/显存带宽性能不足.
获取内存_如何获取一个进程所占用的内存
推荐观看: BATJ面试官最喜欢问的:多线程.线程并发面试题详解(volatile+ThreadLocal+Sleep)_哔哩哔哩 (゜-゜)つロ干杯~-bilibiliwww.bilibili. ...
C#获取特定进程CPU和内存使用率
C#获取特定进程CPU和内存使用率首先是获取特定进程对象,可以使用Process.GetProcesses()方法来获取系统中运行的所有进程,或者使用Process.GetCurrentProces ...
矩池云中Tensorflow指定GPU及GPU显存设置
矩池云中Tensorflow指定GPU及GPU显存设置指定GPU 查看机器上GPU情况命令: nvidia-smi 功能:显示机器上gpu的情况命令: nvidia-smi -l 功能:定时更新 ...
keras系列︱keras是如何指定显卡且限制显存用量（GPU/CPU使用）
keras在使用GPU的时候有个特点,就是默认全部占满显存. 若单核GPU也无所谓,若是服务器GPU较多,性能较好,全部占满就太浪费了. 于是乎有以下五种情况: 1.指定GPU 2.使用固定显存的GP ...
32位应用程序单个进程最大占用内存是4GB
32位应用程序单个进程最大占用内存是4GB左右,这个问题是因为需要测试大数据里才能体现出来的问题,所以反反复复花了一周多的时间来搞定它,希望大家在对它有一个印象.64位的应用程序的时候就不会有这个限制 ...
nvidia-smi 显示无进程，但GPU显存被占用
问题场景训练网络时,未等网络训练完,中途按了:ctrl + c结束网络训练.当再次进行训练网络时,提示GPU显存不足,使用nvidia-smi查看GPU,无进程占用,但GPU显存被占满. 问题分析 ...
查看java进程占用内存_如何查看java进程大批占用内存
如何查看java进程大量占用内存你好,方法如下: 可以直接使用top命令后,查看%MEM的内容.可以选择按进程查看或者按用户查看,如想查看oracle用户的进程内存使用情况的话可以使用如下的命令: ...
Linux--无进程却显示占用显存，GPU显存释放方法
文章目录一.用nvidia-smi查看对应的进程pid 二.用nvidia-smi查不到对应的进程pid 参考链接: Linux–无进程却显示占用显存,GPU显存释放方法程序结束掉,但GPU显存没 ...

获取指定进程所占用内存、指定GPU显存信息、所有GPU型号和驱动

获取指定进程所占用内存、指定GPU显存信息、所有GPU型号和驱动相关推荐

最新文章

热门文章