update

app.py

@@ -1,47 +1,5 @@
-import os
-import sys
-import torch
 import subprocess
 
-def install_pytorch3d():
-    try:
-        import pytorch3d
-        print("✅ PyTorch3D already installed.")
-        return
-    except ImportError:
-        print("⏳ PyTorch3D not found. Starting dynamic installation...")
-
-    pyt_version_str = torch.__version__.split("+")[0].replace(".", "")
-    version_str = "".join([
-        f"py3{sys.version_info.minor}_",
-        f"cu{torch.version.cuda.replace('.', '')}_",
-        f"pyt{pyt_version_str}"
-    ])
-    
-    whl_url = f"https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/pytorch3d-0.7.8-cp3{sys.version_info.minor}-cp3{sys.version_info.minor}-linux_x86_64.whl"
-    
-    print(f"🔍 Detected Env: {version_str}")
-    print(f"⬇️  Attempting to install from Wheel: {whl_url}")
-
-    try:
-        subprocess.run([sys.executable, "-m", "pip", "install", whl_url], check=True)
-        print("✅ PyTorch3D installed via Wheel!")
-    except subprocess.CalledProcessError:
-        print("⚠️  Wheel installation failed (maybe version mismatch).")
-        print("🔨 Falling back to source compilation (this will take a few minutes)...")
-        
-        subprocess.run(
-            [sys.executable, "-m", "pip", "install", "--no-build-isolation", "git+https://github.com/facebookresearch/pytorch3d.git@stable"], 
-            check=True
-        )
-        print("✅ PyTorch3D installed via Source Build!")
-
-# --- 执行安装函数 ---
-install_pytorch3d()
-
-# --- 下面才是你原本的 imports ---
-# from unish.pipeline import UniSHPipeline ...
-
 import gradio as gr
 import os
 import time
@@ -88,6 +46,78 @@ def prepare_local_assets():
 
 prepare_local_assets()
 
+def install_pytorch3d():
+    try:
+        import pytorch3d
+        print("✅ PyTorch3D already installed.")
+        return
+    except ImportError:
+        print("⏳ PyTorch3D not found. Starting dynamic installation...")
+
+    pyt_version_str = torch.__version__.split("+")[0].replace(".", "")
+    version_str = "".join([
+        f"py3{sys.version_info.minor}_",
+        f"cu{torch.version.cuda.replace('.', '')}_",
+        f"pyt{pyt_version_str}"
+    ])
+    
+    whl_url = f"https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/{version_str}/pytorch3d-0.7.8-cp3{sys.version_info.minor}-cp3{sys.version_info.minor}-linux_x86_64.whl"
+    
+    print(f"🔍 Detected Env: {version_str}")
+    print(f"⬇️  Attempting to install from Wheel: {whl_url}")
+
+    try:
+        subprocess.run([sys.executable, "-m", "pip", "install", whl_url], check=True)
+        print("✅ PyTorch3D installed via Wheel!")
+    except subprocess.CalledProcessError:
+        print("⚠️  Wheel installation failed (maybe version mismatch).")
+        print("🔨 Falling back to source compilation (this will take a few minutes)...")
+        
+        subprocess.run(
+            [sys.executable, "-m", "pip", "install", "--no-build-isolation", "git+https://github.com/facebookresearch/pytorch3d.git@stable"], 
+            check=True
+        )
+        print("✅ PyTorch3D installed via Source Build!")
+
+install_pytorch3d()
+
+def install_mmcv():
+    try:
+        import mmcv
+        print(f"✅ mmcv {mmcv.__version__} is already installed.")
+        return
+    except ImportError:
+        print("⏳ mmcv not found. Starting dynamic installation...")
+
+    # 1. Detect environment versions to construct the URL dynamically
+    # Example: CUDA 12.1 -> "121", Torch 2.4.0 -> "2.4"
+    cuda_ver = torch.version.cuda.replace(".", "")
+    torch_ver = ".".join(torch.__version__.split(".")[:2])
+    
+    # 2. Construct the find-links URL matching OpenMMLab's structure
+    # Structure: https://download.openmmlab.com/mmcv/dist/cu{CUDA}/torch{TORCH}/index.html
+    find_links_url = f"https://download.openmmlab.com/mmcv/dist/cu{cuda_ver}/torch{torch_ver}/index.html"
+    
+    print(f"🔍 Detected Env: CUDA={cuda_ver}, Torch={torch_ver}")
+    print(f"⬇️  Installing mmcv==2.2.0 from: {find_links_url}")
+
+    try:
+        # 3. Run pip install with the specific version and dynamic link
+        subprocess.run([
+            sys.executable, "-m", "pip", "install", 
+            "mmcv==2.2.0", 
+            "--find-links", find_links_url
+        ], check=True)
+        print("✅ mmcv installed successfully.")
+        
+    except subprocess.CalledProcessError:
+        print("⚠️  Installation failed. The specific version might not exist for this environment.")
+        print("🔄 Attempting fallback using openmim (auto-resolve mode)...")
+        
+        # Fallback: Install openmim and let it handle the resolution
+        subprocess.run([sys.executable, "-m", "pip", "install", "openmim"], check=True)
+        subprocess.run(["mim", "install", "mmcv==2.2.0"], check=True)
+
 def install_sam2():
     try:
         import sam2

requirements.txt

@@ -20,6 +20,5 @@ git+https://github.com/EasternJournalist/utils3d.git@3fab839f0be9931dac7c8488eb0
 mmcv==2.2.0
 fvcore
 iopath
-git+https://github.com/facebookresearch/pytorch3d.git@stable
 smplx
 hydra-core