From fa456cdc3eb5a70f67cb3d9dbf0e14d8a4b24901 Mon Sep 17 00:00:00 2001 From: mayukhdeb Date: Fri, 24 Oct 2025 08:58:01 -0400 Subject: [PATCH] fall back to CPU if no GPUs are available --- encoding/models/nested_cv.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/encoding/models/nested_cv.py b/encoding/models/nested_cv.py index afa0d97..0d38fac 100644 --- a/encoding/models/nested_cv.py +++ b/encoding/models/nested_cv.py @@ -87,6 +87,9 @@ def fit_predict( device = "mps:0" elif torch.cuda.is_available(): device = "cuda" + else: + # if no GPU is available, fall back to CPU + device = "cpu" else: device = "cpu" logger.info(f"Using device: {device}")