handle when cuda devices in powered down state

tobybreckon · Jan 13, 2025 · bd9fa53 · bd9fa53
1 parent 73cc61c
commit bd9fa53
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/version.py b/version.py
@@ -88,10 +88,14 @@ def non_free_opencv_algorithms_present():
 
 #####################################################################
 
-for gpu in range(cv2.cuda.getCudaEnabledDeviceCount()):
-    print("CUDA enabled GPU device index: " + str(gpu) + " ")
-    cv2.cuda.printShortCudaDeviceInfo(gpu)
-    print()
+try:
+    for gpu in range(cv2.cuda.getCudaEnabledDeviceCount()):
+        print("CUDA enabled GPU device index: " + str(gpu) + " ")
+        cv2.cuda.printShortCudaDeviceInfo(gpu)
+        print()
+except BaseException:
+    print("No CUDA enabled devices found : " +
+          "[none present or in powered down state]")
 
 #####################################################################