Update utils/fusion.py

utils/fusion.py

@@ -5,19 +5,19 @@ import torch
 from numba import njit, prange
 from skimage import measure
 
-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
+# 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
 
 class TSDFVolume:
   """Volumetric TSDF Fusion of RGB-D Images.
@@ -57,6 +57,7 @@ class TSDFVolume:
 
     #self.gpu_mode = False # CPU for debugging!!
     self.gpu_mode = use_gpu and FUSION_GPU_MODE
+    print('GPU Fusion Mode: ',self.gpu_mode)
 
     # Copy voxel volumes to GPU
     if self.gpu_mode: