Don't crash if nvidia-smi returns an error

2025-02-05 11:40:04 -05:00 · 2012-01-18 11:36:12 -05:00 · 2012-01-18 11:36:12 -05:00 · 8c3a698596
commit 8c3a698596
parent 1ff0deb16b
1 changed files with 4 additions and 1 deletions
--- a/worker.py
+++ b/worker.py
@ -37,7 +37,10 @@ uu = lambda t: ':'.join((t, uuid.uuid1().hex))

 def get_temperature():
    id = pycuda.autoinit.device.pci_bus_id()
+    try:
        out = check_output('nvidia-smi -q -d TEMPERATURE'.split())
+    except:
+        return ''
    idx = out.find('\nGPU ' + id)
    if idx >= 0:
        out.find('Gpu', idx)