Update app.py

app.py

@@ -13,32 +13,29 @@ from point_cloud_generator import display_pcd
 
 
 
+import torch
 import subprocess
+import os
+
+# Check if CUDA is available in PyTorch
+print(f"🚀 PyTorch CUDA Available: {torch.cuda.is_available()}")
 
-# Check if CUDA is available
-def check_gpu():
-    try:
-        # Run nvidia-smi to check for GPU
-        gpu_check = subprocess.run(["nvidia-smi"], capture_output=True, text=True)
-        if "NVIDIA-SMI" in gpu_check.stdout:
-            print("✅ GPU is available!")
-            return True
-        else:
-            print("❌ No GPU detected!")
-            return False
-    except FileNotFoundError:
-        print("❌ No nvidia-smi found! GPU likely unavailable.")
-        return False
-
-# Set device based on GPU availability
-if check_gpu() and torch.cuda.is_available():
-    print(f"✅ CUDA available: {torch.cuda.get_device_name(0)}")
-    device = torch.device("cuda")
-    torch.cuda.empty_cache()
+# Check available CUDA devices
+if torch.cuda.is_available():
+    print(f"✅ GPU detected: {torch.cuda.get_device_name(0)}")
 else:
-    print("❌ No CUDA available. Falling back to CPU.")
-    os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
-    device = torch.device("cpu")
+    print("❌ No GPU detected by PyTorch!")
+
+# Check `nvidia-smi` output (if NVIDIA GPU is assigned)
+try:
+    gpu_check = subprocess.run(["nvidia-smi"], capture_output=True, text=True)
+    print(f"🖥️ nvidia-smi output:\n{gpu_check.stdout}")
+except FileNotFoundError:
+    print("❌ nvidia-smi not found! GPU might not be assigned.")
+
+# Check if Hugging Face assigned a GPU
+hf_gpu_env = os.getenv("CUDA_VISIBLE_DEVICES")
+print(f"🔍 Hugging Face CUDA_VISIBLE_DEVICES: {hf_gpu_env}")