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
