update

app.py

@@ -426,7 +426,7 @@ def main(args_1, args_2):
         )
 
     demo.queue(max_size=1)
-    demo.launch(share=True)
+    demo.launch(share=False)
 
 if __name__ == "__main__":
 

nsr/lsgm/flow_matching_trainer.py

@@ -126,7 +126,8 @@ from utils.general_utils import matrix_to_quaternion
 from utils.mesh_util import post_process_mesh, to_cam_open3d_compat
 from datasets.g_buffer_objaverse import focal2fov, fov2focal
 
-import open3d as o3d
+# import open3d as o3d
+import open3d_zerogpu_fix
 
 # from sgm.sampling_utils.demo.streamlit_helpers import init_sampling
 

nsr/train_nv_util.py

@@ -28,7 +28,8 @@ import torch
 from torch.autograd import Function
 import torch.nn.functional as F
 import torch as th
-import open3d as o3d
+# import open3d as o3d
+import open3d_zerogpu_fix
 import torch.distributed as dist
 import torchvision
 from PIL import Image

open3d_zerogpu_fix.py

@@ -0,0 +1,7 @@
+import fileinput
+import site
+from pathlib import Path
+
+with fileinput.FileInput(f'{site.getsitepackages()[0]}/open3d/__init__.py', inplace=True) as file:
+    for line in file:
+        print(line.replace('_pybind_cuda.open3d_core_cuda_device_count()', '1'), end='')

utils/mesh_util.py

@@ -13,7 +13,8 @@ import cv2
 import numpy as np
 from PIL import Image
 from functools import partial
-import open3d as o3d
+# import open3d as o3d
+import open3d_zerogpu_fix
 import trimesh