GPUtil-在Python中获取GPU状态

code • 2022-5-5 • java • 阅读 38

简介

GPUtil是一个可以在Python中使用nvidia-smi从NVIDA GPU获取GPU状态的第三方库。

使用

# 安装
pip install gputil
# 使用
import GPUtil as GPU
import sys
'''
#  *** 作系统
print('OS: ' + sys.platform)
# python版本
print(sys.version)
# GPUtil版本
print(GPU.__name__ + ' ' + GPU.__version__)
'''

Gpus = GPU.getGPUs()
for gpu in Gpus:
    print