#14199: catch subprocess.CalledProcessError in get_gpus()

apache · Feb 20, 2019 · ceb5c25 · ceb5c25
1 parent f9c436b
commit ceb5c25
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/example/image-classification/common/util.py b/example/image-classification/common/util.py
@@ -51,6 +51,6 @@ def get_gpus():
     """
     try:
         re = subprocess.check_output(["nvidia-smi", "-L"], universal_newlines=True)
-    except OSError:
+    except (subprocess.CalledProcessError, OSError):
         return []
     return range(len([i for i in re.split('\n') if 'GPU' in i]))