From 4c1daf9be43846509418577737aa37419e1a7aef Mon Sep 17 00:00:00 2001 From: sronilsson Date: Fri, 22 Nov 2024 13:31:14 -0500 Subject: [PATCH] env --- simba/data_processors/cuda/image.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/simba/data_processors/cuda/image.py b/simba/data_processors/cuda/image.py index 5380e4619..500c68b8f 100644 --- a/simba/data_processors/cuda/image.py +++ b/simba/data_processors/cuda/image.py @@ -19,7 +19,10 @@ import cv2 import numpy as np -from cupyx.scipy.ndimage import rotate +try: + from cupyx.scipy.ndimage import rotate +except: + from scipy.ndimage import rotate from numba import cuda from simba.data_processors.cuda.utils import _cuda_mse