catch all exceptions when importing cupy

TNTwise · Dec 8, 2024 · 1e28203 · 1e28203
1 parent ac5d942
commit 1e28203
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/src/constants.py b/backend/src/constants.py
@@ -5,7 +5,7 @@ def checkForCUDA() -> bool:
         import torch
         import torchvision
         import cupy
-    except ImportError as e:
+    except Exception as e:
         return False
     if cupy.cuda.get_cuda_path() == None:
         return False