Python 查看顯存大小

背景：由於一些需求，須要在python程序中查看顯存的佔用狀況。 1.安裝NVML庫 pip install nvidia-ml-pypython 2.使用 import pynvml pynvml.nvmlInit() handle = pynvml.nvmlDeviceGetHandleByIndex(0) #gpu_id meminfo = pynvml.nvmlDeviceGetMem

>>阅读原文<<