fix: use prebuilt wheel files.

app.py

@@ -87,20 +87,20 @@ def setup_runtime_env():
     logging.info("CUDA is available: %s" % torch.cuda.is_available())
     logging.info("CUDA Device Capability: %s" % (torch.cuda.get_device_capability(),))
 
-    # Install Pre-compiled CUDA extensions (Not working)
+    # Install Pre-compiled CUDA extensions (Fallback to this solution on 12/31/24)
     # Ref: https://huggingface.co/spaces/zero-gpu-explorers/README/discussions/110
-    #
-    # ext_dir = os.path.join(os.path.dirname(__file__), "wheels")
-    # for e in os.listdir(ext_dir):
-    #     logging.info("Installing Extensions from %s" % e)
-    #     subprocess.call(
-    #         ["pip", "install", os.path.join(ext_dir, e)], stderr=subprocess.STDOUT
-    #     )
-    # Compile CUDA extensions
-    ext_dir = os.path.join(os.path.dirname(__file__), "citydreamer", "extensions")
+    ext_dir = os.path.join(os.path.dirname(__file__), "wheels")
     for e in os.listdir(ext_dir):
-        if os.path.isdir(os.path.join(ext_dir, e)):
-            subprocess.call(["pip", "install", "."], cwd=os.path.join(ext_dir, e))
+        logging.info("Installing Extensions from %s" % e)
+        subprocess.call(
+            ["pip", "install", os.path.join(ext_dir, e)], stderr=subprocess.STDOUT
+        )
+    # Compile CUDA extensions
+    # Update on 12/31/24: No module named 'torch'. But it is installed and listed by `pip list`
+    # ext_dir = os.path.join(os.path.dirname(__file__), "citydreamer", "extensions")
+    # for e in os.listdir(ext_dir):
+    #     if os.path.isdir(os.path.join(ext_dir, e)):
+    #         subprocess.call(["pip", "install", "."], cwd=os.path.join(ext_dir, e))
 
     logging.info("Installed Python Packages: %s" % _get_output(["pip", "list"]))
 
@@ -183,13 +183,13 @@ if __name__ == "__main__":
         format="[%(levelname)s] %(asctime)s %(message)s", level=logging.INFO
     )
     logging.info("Environment Variables: %s" % os.environ)
-    if _get_output(["nvcc", "--version"]) is None:
-        logging.info("Installing CUDA toolkit...")
-        install_cuda_toolkit()
-    else:
-        logging.info("Detected CUDA: %s" % _get_output(["nvcc", "--version"]))
+    # if _get_output(["nvcc", "--version"]) is None:
+    #     logging.info("Installing CUDA toolkit...")
+    #     install_cuda_toolkit()
+    # else:
+    #     logging.info("Detected CUDA: %s" % _get_output(["nvcc", "--version"]))
 
-    logging.info("Compiling CUDA extensions...")
+    logging.info("Installing CUDA extensions...")
     setup_runtime_env()
 
     logging.info("Downloading pretrained models...")

wheels/extrude_tensor-1.0.0-cp310-cp310-linux_x86_64.whl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ed313f42a9ded9fe5fb3a73fdb2218fa71fde4ced00fc76e027cba24faf5ad34
+size 67708

wheels/grid_encoder-1.0.0-cp310-cp310-linux_x86_64.whl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cd18d20dcf65f5895280b04f77dbca52bca473ffff1b87d8f1aadfaf4cf91e97
+size 2345077

wheels/voxrender-1.0.0-cp310-cp310-linux_x86_64.whl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:28751d39899af5bc1db6905405abf92a8b72f5701331584fe794d688ba037ee1
+size 593621