nvidia-ml-py安装 版本:375.53.1 pip install nvidia-ml-py 使用 import pynvml # 初始化模块 pynvml.nvmlInit() # 获取0号位GPU gpu = punvml.nvmlDeviceGeHandleByIndex(0) gpu_info = pynvml.nvmlDeviceGetMemoryInfo(gpu) 获取的gpu_info有三个字段信息,获取方式如下 gpu_info