Fix CUDA to 12.2.

README.md

@@ -15,5 +15,8 @@ Official demo for **[CityDreamer: Compositional Generative Model of Unbounded 3D
 - 🤗 Try CityDreamer to generate photolistic 3D cities.
 - ⚠️ Due to the limited computational resources at Hugging Face, this demo only generates **A SINGLE IMAGE** based on the New York City layout. If you wish to experience more comprehensive functionality, please visit the demo on [GitHub](https://github.com/hzxie/city-dreamer?tab=readme-ov-file#iterative-demo-%EF%B8%8F).
 
-<p style="color:red">❕IMPORTANT NOTE: The HF team and we are currently working on migrating to ZeroGPU, so the demo may be temporarily unavailable.</p>
+<p style="color:red">
+❕IMPORTANT NOTE: The HF team and we are currently working on migrating to ZeroGPU, so the demo may be temporarily unavailable.<br>
+Track the debugging progress on <a href="https://huggingface.co/spaces/zero-gpu-explorers/README/discussions/111">the community of zero-gpu-explorers</a>.
+</p>
 

app.py

@@ -4,7 +4,7 @@
 # @Author: Haozhe Xie
 # @Date:   2024-03-02 16:30:00
 # @Last Modified by: Haozhe Xie
-# @Last Modified at: 2024-09-22 15:23:37
+# @Last Modified at: 2024-09-22 21:11:51
 # @Email:  [email protected]
 
 import gradio as gr
@@ -12,15 +12,39 @@ import logging
 import numpy as np
 import os
 
-import spaces
 import ssl
 import subprocess
 import sys
-import torch
 import urllib.request
 
 from PIL import Image
 
+# Reinstall PyTorch with CUDA 11.8 (Default version is 12.1)
+# subprocess.call(
+#     [
+#         "pip",
+#         "install",
+#         "torch==2.2.2",
+#         "torchvision==0.17.2",
+#         "--index-url",
+#         "https://download.pytorch.org/whl/cu118",
+#     ]
+# )
+import torch
+
+# Create a dummy decorator for Non-ZeroGPU environments
+if os.environ.get("SPACES_ZERO_GPU") is not None:
+    import spaces
+else:
+    class spaces:
+        @staticmethod
+        def GPU(func):
+            # This is a dummy wrapper that just calls the function.
+            def wrapper(*args, **kwargs):
+                return func(*args, **kwargs)
+
+            return wrapper
+
 # Fix: ssl.SSLCertVerificationError: [SSL: CERTIFICATE_VERIFY_FAILED] certificate verify failed
 ssl._create_default_https_context = ssl._create_unverified_context
 # Import CityDreamer modules
@@ -43,7 +67,7 @@ def install_cuda_toolkit():
     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" % (
@@ -61,7 +85,9 @@ 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
+    # Install Pre-compiled CUDA extensions (Not working)
+    # 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)
@@ -154,6 +180,7 @@ if __name__ == "__main__":
     logging.basicConfig(
         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()

requirements.txt

@@ -1,8 +1,7 @@
-torch==2.2.2
-torchvision==0.17.2
-
 easydict
 gradio
 numpy<2.0.0
 opencv-python
 pillow
+torch==2.2.2
+torchvision==0.17.2