remove installation

app.py

@@ -29,41 +29,41 @@ import spaces
 from spaces import zero
 zero.startup()
 
-def install_cuda_toolkit():
-    # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
-    # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
-    CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run"
-    CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
-    subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
-    subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
-    subprocess.call([CUDA_TOOLKIT_FILE, "--silent", "--toolkit"])
-
-    os.environ["CUDA_HOME"] = "/usr/local/cuda"
-    os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])
-    os.environ["LD_LIBRARY_PATH"] = "%s/lib:%s" % (
-        os.environ["CUDA_HOME"],
-        "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
-    )
-    # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
-    os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
-
-install_cuda_toolkit()
-
-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")
-
-subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-deps", "dgl", "-f", "https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html"])
+# def install_cuda_toolkit():
+#     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run"
+#     # CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run"
+#     CUDA_TOOLKIT_URL = "https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run"
+#     CUDA_TOOLKIT_FILE = "/tmp/%s" % os.path.basename(CUDA_TOOLKIT_URL)
+#     subprocess.call(["wget", "-q", CUDA_TOOLKIT_URL, "-O", CUDA_TOOLKIT_FILE])
+#     subprocess.call(["chmod", "+x", CUDA_TOOLKIT_FILE])
+#     subprocess.call([CUDA_TOOLKIT_FILE, "--silent", "--toolkit"])
+
+#     os.environ["CUDA_HOME"] = "/usr/local/cuda"
+#     os.environ["PATH"] = "%s/bin:%s" % (os.environ["CUDA_HOME"], os.environ["PATH"])
+#     os.environ["LD_LIBRARY_PATH"] = "%s/lib:%s" % (
+#         os.environ["CUDA_HOME"],
+#         "" if "LD_LIBRARY_PATH" not in os.environ else os.environ["LD_LIBRARY_PATH"],
+#     )
+#     # Fix: arch_list[-1] += '+PTX'; IndexError: list index out of range
+#     os.environ["TORCH_CUDA_ARCH_LIST"] = "8.0;8.6"
+
+# install_cuda_toolkit()
+
+# 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")
+
+# subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-deps", "dgl", "-f", "https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html"])
 
 # subprocess.check_call([sys.executable, "-m", "pip", "uninstall", "-y", "torch", "torchvision", "torchaudio"])
 # subprocess.check_call([sys.executable, "-m", "pip", "install", "torch==2.4.0", "torchvision==0.19.0", "torchaudio==2.4.0", "--index-url", "https://download.pytorch.org/whl/cu124"])
 # os.system('conda install conda-forge::ffmpeg')
 
-site.main()  # re-processes every *.pth in site-packages
-importlib.invalidate_caches()
-torch = importlib.import_module("torch")
+# site.main()  # re-processes every *.pth in site-packages
+# importlib.invalidate_caches()
+# torch = importlib.import_module("torch")
 
 import torch
 import torch.nn as nn

requirements.txt

@@ -1,4 +1,4 @@
-warp-lang==1.1.0
+# warp-lang==1.1.0
 trimesh
 pyyaml
 numpy==1.26.4
@@ -23,6 +23,27 @@ timm
 omegaconf
 hydra-core
 wandb
-torch==2.6.0 --index-url https://download.pytorch.org/whl/cu124
-torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
-torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
+
+# torch==2.6.0 --index-url https://download.pytorch.org/whl/cu124
+# torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
+# torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
+# diff-gaussian-rasterization @ file://./src/third-party/diff-gaussian-rasterization-w-depth
+# dgl==2.4.0+cu124     --find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
+
+# --- core CUDA wheels -------------------------------------------------
+--extra-index-url https://download.pytorch.org/whl/cu124
+torch==2.4.0+cu124          # ⇦ install Torch first
+torchvision==0.19.0+cu124
+torchaudio==2.4.0+cu124
+
+# --- DGL (needs the matching Torch 2.4 wheel) -------------------------
+--find-links https://data.dgl.ai/wheels/torch-2.4/cu124/repo.html
+dgl==2.4.0+cu124
+
+warp-lang==1.1.0
+
+# --- everything that depends on torch *after* this point --------------
+# editable build (keeps recompiling when you edit the code)
+-e ./src/third-party/diff-gaussian-rasterization-w-depth
+# or, if you prefer an immutable wheel:
+# diff-gaussian-rasterization @ ./src/third-party/diff-gaussian-rasterization-w-depth

src/experiments/real_world/gs/helpers.py

@@ -3,9 +3,9 @@ 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")
+# import importlib
+# diff_gaussian_rasterization = importlib.import_module("diff_gaussian_rasterization")
 from diff_gaussian_rasterization import GaussianRasterizationSettings as Camera