debug

app.py

@@ -181,6 +181,12 @@ def text_to_detailed(prompt, seed=None):
     os.environ["TORCH_CUDA_ARCH_LIST"] = "9.0"
     if "CUDA_VISIBLE_DEVICES" in os.environ:
         del os.environ["CUDA_VISIBLE_DEVICES"]
+    os.environ['CUDA_HOME'] = '/usr/local/cuda-12.1'
+    os.environ['PATH'] += ':/usr/local/cuda-12.1/bin'
+    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()) 
     subprocess.run(['nvidia-smi'])       
     import nvdiffrast.torch as dr
     dr.RasterizeCudaContext()