Update utils/fusion.py

utils/fusion.py

@@ -5,15 +5,19 @@ import torch
 from numba import njit, prange
 from skimage import measure
 
-if torch.cuda.is_available():
-  import pycuda.driver as cuda
-  import pycuda.autoinit
-  from pycuda.compiler import SourceModule
-  FUSION_GPU_MODE = 1
-else:
+try:
+    import pycuda.driver as cuda
+    cuda.init()
+    device_count = cuda.Device.count()
+    if device_count > 0:
+        import pycuda.autoinit
+        from pycuda.compiler import SourceModule
+        FUSION_GPU_MODE = 1
+    else:
+        raise RuntimeError("No CUDA devices found.")
+except Exception as e:
+    print(f"[!] CUDA init failed or not available: {e}")
     FUSION_GPU_MODE = 0
-    print("CUDA not available, running on CPU.")
-
 
 class TSDFVolume:
   """Volumetric TSDF Fusion of RGB-D Images.