Check also for OSError in fftp.py. Should fix #21

vincefn · Nov 26, 2022 · 66fe944 · 66fe944
1 parent fe7bafa
commit 66fe944
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyvkfft/fft.py b/pyvkfft/fft.py
@@ -21,14 +21,14 @@
         import pycuda.gpuarray as cua
     if has_cupy:
         import cupy as cp
-except ImportError:
+except (ImportError, OSError):
     has_cupy, has_pycuda = False, False
 
 try:
     from .opencl import VkFFTApp as VkFFTApp_cl, cla, vkfft_version
 
     has_opencl = True
-except ImportError:
+except (ImportError, OSError):
     has_opencl = False