debug

app.py

@@ -67,8 +67,8 @@ if not UI_ONLY_MODE:
 
 @spaces.GPU
 def check_gpu():
-    if "CUDA_VISIBLE_DEVICES" in os.environ:
-        del os.environ["CUDA_VISIBLE_DEVICES"]
+    # 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', '')
@@ -176,7 +176,8 @@ def text_to_detailed(prompt, seed=None):
         shlex.split(
             "rm -rf /home/user/.cache/torch_extensions/py310_cu121/nvdiffrast_plugin/lock"
         )
-    )       
+    )
+    subprocess.run(['nvidia-smi'])       
     import nvdiffrast.torch as dr
     dr.RasterizeCudaContext(device="cuda:0")
     print("nvdiffrast initialized successfully")