diff --git a/build/hooks/hook-utils.gpu.py b/build/hooks/hook-utils.gpu.py index b34bcea..23e4027 100644 --- a/build/hooks/hook-utils.gpu.py +++ b/build/hooks/hook-utils.gpu.py @@ -2,4 +2,12 @@ from utils.gpu import CUDA_PACKAGES -hiddenimports = [pkg for pkg in CUDA_PACKAGES if importlib.util.find_spec(pkg) is not None] + +def _is_importable(pkg: str) -> bool: + try: + return importlib.util.find_spec(pkg) is not None + except ModuleNotFoundError: + return False + + +hiddenimports = [pkg for pkg in CUDA_PACKAGES if _is_importable(pkg)]