debug

app.py

@@ -74,7 +74,12 @@ def check_gpu():
     os.environ['LD_LIBRARY_PATH'] = "/usr/local/cuda-12.1/lib64:" + os.environ.get('LD_LIBRARY_PATH', '')
     subprocess.run(['nvidia-smi'])  # Test if CUDA is available
     print(f"torch.cuda.is_available:{torch.cuda.is_available()}")
-    print("Device count:", torch.cuda.device_count())
+    print("Device count:", torch.cuda.device_count())        
+    # test nvdiffrast
+    import nvdiffrast.torch as dr
+    dr.RasterizeCudaContext(device="cuda:0")
+    print("nvdiffrast initialized successfully")
+    
 
 # Only check GPU in non-UI debug mode
 if not UI_ONLY_MODE:
@@ -187,7 +192,7 @@ else:
         import nvdiffrast.torch as dr
         dr.RasterizeCudaContext(device="cuda:0")
         print("nvdiffrast initialized successfully")
-        
+
         print(f"torch.cuda.is_available():{torch.cuda.is_available()}")
         # print(f"Before text_to_detailed: {torch.cuda.memory_allocated() / 1024**3} GB")
         return k3d_wrapper.get_detailed_prompt(prompt, seed)