mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-02-05 11:40:04 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Don't crash if nvidia-smi returns an error

Browse Source
This commit is contained in:
Steven Robertson 2012-01-18 11:36:12 -05:00
parent 1ff0deb16b
commit 8c3a698596
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
worker.py
View File

@ -37,7 +37,10 @@ uu = lambda t: ':'.join((t, uuid.uuid1().hex))
def get_temperature(): def get_temperature():
id = pycuda.autoinit.device.pci_bus_id() id = pycuda.autoinit.device.pci_bus_id()
try:
out = check_output('nvidia-smi -q -d TEMPERATURE'.split()) out = check_output('nvidia-smi -q -d TEMPERATURE'.split())
except:
return ''
idx = out.find('\nGPU ' + id) idx = out.find('\nGPU ' + id)
if idx >= 0: if idx >= 0:
out.find('Gpu', idx) out.find('Gpu', idx)