debug

app.py

@@ -77,6 +77,7 @@ def check_gpu():
     print("Device count:", torch.cuda.device_count()) 
 
     # test nvdiffrast
+    os.environ["TORCH_CUDA_ARCH_LIST"] = "9.0"
     import nvdiffrast.torch as dr
     dr.RasterizeCudaContext(device="cuda:0")
     print("nvdiffrast initialized successfully")
@@ -177,6 +178,7 @@ def text_to_detailed(prompt, seed=None):
             "rm -rf /home/user/.cache/torch_extensions/py310_cu121/nvdiffrast_plugin/lock"
         )
     )
+    os.environ["TORCH_CUDA_ARCH_LIST"] = "9.0"
     if "CUDA_VISIBLE_DEVICES" in os.environ:
         del os.environ["CUDA_VISIBLE_DEVICES"]
     subprocess.run(['nvidia-smi'])