Skip to content
Snippets Groups Projects
Commit 253d5bc9 authored by Chris Hines's avatar Chris Hines
Browse files

more gpu metrics being collected

Former-commit-id: 8ae7e976
parent 4b8fd797
No related branches found
No related tags found
No related merge requests found
#!/usr/bin/env python
import collectd
try:
import collectd
from collectd import Values
except:
from fakeCollectd import Values
import subprocess
import xml.etree.ElementTree as ET
def read(data=None):
vl = collectd.Values(type='gauge')
vl = Values(type='gauge')
vl.plugin = 'cuda'
out = subprocess.check_output(['nvidia-smi', '-q', '-x'])
......@@ -30,6 +35,18 @@ def read(data=None):
values=[1e6 * float(gpu.find('utilization/gpu_util').text.split()[0])])
except:
pass
collectd.register_read(read)
try:
vl.dispatch(type='gauge', type_instance='power_state',
values=[int(gpu.find('power_readings/power_state').text.split()[0][1:])])
except:
pass
try:
vl.dispatch(type='gauge', type_instance='power_draw',
values=[float(gpu.find('power_readings/power_draw').text.split()[0])])
except:
pass
try:
collectd.register_read(read)
except:
readtest()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment