update

app.py

@@ -1,4 +1,7 @@
 import gradio as gr
+import importlib
+import sys
+import site
 from PIL import Image
 from pathlib import Path
 from omegaconf import DictConfig, OmegaConf
@@ -46,8 +49,13 @@ def install_cuda_toolkit():
     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
 
 install_cuda_toolkit()
-gs_path = str(Path(__file__).parent / 'src/third-party/diff-gaussian-rasterization-w-depth')
-os.system(f'pip install -e {gs_path}')
+
+gs_path = Path(__file__).parent / "src/third-party/diff-gaussian-rasterization-w-depth"
+subprocess.check_call([sys.executable, "-m", "pip", "install", "-e", str(gs_path)])
+site.main()  # re-processes every *.pth in site-packages
+importlib.invalidate_caches()
+diff_gaussian_rasterization = importlib.import_module("diff_gaussian_rasterization")
+
 os.system('pip install  dgl -f https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html')
 # os.system('conda install conda-forge::ffmpeg')
 

src/experiments/real_world/gs/helpers.py

@@ -2,6 +2,10 @@ import torch
 import os
 import open3d as o3d
 import numpy as np
+from pathlib import Path
+import importlib
+
+diff_gaussian_rasterization = importlib.import_module("diff_gaussian_rasterization")
 from diff_gaussian_rasterization import GaussianRasterizationSettings as Camera