diff --git a/setup.py b/setup.py index dc5d496..6a417b3 100644 --- a/setup.py +++ b/setup.py @@ -80,7 +80,7 @@ def get_compiler(): extra_compile_args={'nvcc': ['-O2']}, ), ) -except ImportError: +except (ImportError, OSError): warnings.warn("Cuda extensions for torch not found, skipping cuda healpix padding module")