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', '')
@@ -172,11 +172,13 @@ def save_py3dmesh_with_trimesh_fast(meshes, save_glb_path=TEMP_MESH_ADDRESS, app
 @spaces.GPU
 def text_to_detailed(prompt, seed=None):
     # test nvdiffrast
-    subprocess.run(
-        shlex.split(
-            "rm -rf /home/user/.cache/torch_extensions/py310_cu121/nvdiffrast_plugin/lock"
-        )
-    )
+    # subprocess.run(
+    #     shlex.split(
+    #         "rm -rf /home/user/.cache/torch_extensions/py310_cu121/nvdiffrast_plugin/lock"
+    #     )
+    # )
+    if "CUDA_VISIBLE_DEVICES" in os.environ:
+        del os.environ["CUDA_VISIBLE_DEVICES"]
     subprocess.run(['nvidia-smi'])       
     import nvdiffrast.torch as dr
     dr.RasterizeCudaContext(device="cuda:0")