From 8c3a6985967e92900e99d4d38329f2a94356f83d Mon Sep 17 00:00:00 2001 From: Steven Robertson Date: Wed, 18 Jan 2012 11:36:12 -0500 Subject: [PATCH] Don't crash if nvidia-smi returns an error --- worker.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/worker.py b/worker.py index 48b2355..b82244a 100644 --- 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() - out = check_output('nvidia-smi -q -d TEMPERATURE'.split()) + try: + out = check_output('nvidia-smi -q -d TEMPERATURE'.split()) + except: + return '' idx = out.find('\nGPU ' + id) if idx >= 0: out.find('Gpu', idx)