From 1e282031aef84a2e3b32859acb608737e70905c0 Mon Sep 17 00:00:00 2001 From: tntwise Date: Sat, 7 Dec 2024 23:39:29 -0800 Subject: [PATCH] catch all exceptions when importing cupy --- backend/src/constants.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/src/constants.py b/backend/src/constants.py index 9be1d1e4..333c0e4c 100644 --- 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