查看显存占用大小

一、目录

1. 查看显存

二、实现

1. 查看显存

 import pynvml
pynvml.nvmlInit()
handle = pynvml.nvmlDeviceGetHandleByIndex(0) # 0表示显卡标号
# 在每一个要查看的地方都要重新定义一个meminfo
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.total/1024**2) #总的显存大小
print(meminfo.used/1024**2)  #已用显存大小
print(meminfo.free/1024**2)  #剩余显存大小
# 单位是MB，如果想看G就再除以一个1024