remove cuda runtime check

this prevents the code from building in a docker image.
NVlabs · Sep 11, 2024 · 986881c · 986881c
1 parent 82506cc
commit 986881c
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -74,9 +74,6 @@ def get_compiler():
     import torch
     from torch.utils.cpp_extension import CUDAExtension
 
-    if not torch.cuda.is_available():
-        raise ImportError()
-
     ext_modules.append(
         CUDAExtension(
             name='healpixpad_cuda',