Upload 4 files

post_install.py

@@ -39,10 +39,30 @@ def install_torch_sparse():
     """Install torch-sparse with compatible PyTorch version"""
     print("Installing torch-sparse...")
     
-    # First, install a compatible PyTorch version with CUDA 11.7 (as expected by PyTorch3D)
-    print("Installing compatible PyTorch version...")
-    if not run_command("pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117"):
-        return False
+    # Check if torch-sparse is already installed
+    try:
+        import torch_sparse
+        print("torch-sparse already installed")
+        return True
+    except ImportError:
+        pass
+    
+    # Check if PyTorch is already installed with the correct version
+    try:
+        import torch
+        if torch.__version__.startswith("2.0.1") and "+cu117" in torch.__version__:
+            print(f"PyTorch {torch.__version__} already installed with correct CUDA version")
+        else:
+            print(f"PyTorch {torch.__version__} installed, but need to update to 2.0.1+cu117")
+            # First, install a compatible PyTorch version with CUDA 11.7 (as expected by PyTorch3D)
+            print("Installing compatible PyTorch version...")
+            if not run_command("pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117"):
+                return False
+    except ImportError:
+        # First, install a compatible PyTorch version with CUDA 11.7 (as expected by PyTorch3D)
+        print("Installing compatible PyTorch version...")
+        if not run_command("pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117"):
+            return False
     
     # Check PyTorch installation
     print("Checking PyTorch installation...")
@@ -60,6 +80,14 @@ def install_torch_scatter():
     """Install torch-scatter with compatible PyTorch version"""
     print("Installing torch-scatter...")
     
+    # Check if torch-scatter is already installed
+    try:
+        import torch_scatter
+        print("torch-scatter already installed")
+        return True
+    except ImportError:
+        pass
+    
     # Install torch-scatter with the compatible PyTorch version
     print("Installing torch-scatter with PyTorch 2.0.1...")
     if run_command("pip install torch-scatter -f https://data.pyg.org/whl/torch-2.0.1+cu117.html"):
@@ -72,6 +100,14 @@ def install_nvdiffrast():
     """Install nvdiffrast"""
     print("Installing nvdiffrast...")
     
+    # Check if nvdiffrast is already installed
+    try:
+        import nvdiffrast
+        print("nvdiffrast already installed")
+        return True
+    except ImportError:
+        pass
+    
     # Create packages directory if it doesn't exist
     packages_dir = Path("packages")
     packages_dir.mkdir(exist_ok=True)
@@ -92,6 +128,14 @@ def install_pytorch3d():
     """Install PyTorch3D"""
     print("Installing PyTorch3D...")
     
+    # Check if PyTorch3D is already installed
+    try:
+        import pytorch3d
+        print(f"PyTorch3D already installed: {pytorch3d.__version__}")
+        return True
+    except ImportError:
+        pass
+    
     packages_dir = Path("packages")
     
     # Clone PyTorch3D
@@ -128,16 +172,23 @@ def install_pytorch3d():
         else:
             print("Trying PyTorch3D installation with default CUDA...")
         
-        if run_command("pip install .", cwd=pytorch3d_dir, env=env):
+        # Use --no-build-isolation to allow access to installed torch
+        if run_command("pip install . --no-build-isolation", cwd=pytorch3d_dir, env=env):
             print(f"Successfully installed PyTorch3D with config {i+1}")
             return True
     
     # If all CUDA configs fail, try without FORCE_CUDA
     print("Trying PyTorch3D installation without FORCE_CUDA...")
-    if run_command("pip install .", cwd=pytorch3d_dir):
+    if run_command("pip install . --no-build-isolation", cwd=pytorch3d_dir):
         print("Successfully installed PyTorch3D without CUDA forcing")
         return True
     
+    # If source installation fails, try pre-built wheel
+    print("Trying PyTorch3D installation with pre-built wheel...")
+    if run_command("pip install pytorch3d --no-build-isolation"):
+        print("Successfully installed PyTorch3D from pre-built wheel")
+        return True
+    
     return False
 
 def install_fashion_clip():