########################################################## 获取指定进程所占用内存#########################################################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 idmeminfo = 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()
