pip install pynvml
import pynvml pynvml.nvmlInit() handle = pynvml.nvmlDeviceGetHandleByIndex(0) # 指定GPU的id meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle) print(meminfo.used / 1024**2) print(meminfo.free / 1024**2)
pip install pynvml
import pynvml pynvml.nvmlInit() handle = pynvml.nvmlDeviceGetHandleByIndex(0) # 指定GPU的id meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle) print(meminfo.used / 1024**2) print(meminfo.free / 1024**2)
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
