Python查看GPU已使用的显存

pip安装pynvml

pip install nvidia-ml-py3

如果是Python2，则pip nvidia-ml-py2

之后可以在Python3 下使用了

import pynvml
pynvml.nvmlInit()
# 这里的0是GPU id
handle = pynvml.nvmlDeviceGetHandleByIndex(0)
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.used)

GPU其他信息，可以查看API文档