Python 查看显存大小

背景：由于一些需求，须要在python程序中查看显存的占用状况。 1.安装NVML库 pip install nvidia-ml-pypython 2.使用 import pynvml pynvml.nvmlInit() handle = pynvml.nvmlDeviceGetHandleByIndex(0) #gpu_id meminfo = pynvml.nvmlDeviceGetMem

>>阅读原文<<