Upload 4 files

post_install.py

@@ -66,10 +66,27 @@ def install_torch_sparse():
         
         print(f"PyTorch base version: {pytorch_base}, CUDA: {cuda_version}")
         
-        # Store version info for later use
-        global PYTORCH_VERSION, CUDA_VERSION
-        PYTORCH_VERSION = pytorch_base
-        CUDA_VERSION = cuda_version
+        # Check if we have a very recent PyTorch version that might not have compatible wheels
+        if pytorch_base.startswith("2.7"):
+            print("PyTorch 2.7.x detected - this is very recent and may not have compatible torch-sparse/torch-scatter wheels")
+            print("Attempting to install a more compatible PyTorch version...")
+            
+            # Try to install PyTorch 2.0.1 which has known compatible wheels
+            if run_command("pip install --force-reinstall torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117"):
+                print("Successfully installed PyTorch 2.0.1+cu117")
+                global PYTORCH_VERSION, CUDA_VERSION
+                PYTORCH_VERSION = "2.0.1"
+                CUDA_VERSION = "cu117"
+            else:
+                print("Failed to install PyTorch 2.0.1, will try to work with current version")
+                global PYTORCH_VERSION, CUDA_VERSION
+                PYTORCH_VERSION = pytorch_base
+                CUDA_VERSION = cuda_version
+        else:
+            # Store version info for later use
+            global PYTORCH_VERSION, CUDA_VERSION
+            PYTORCH_VERSION = pytorch_base
+            CUDA_VERSION = cuda_version
         
     except ImportError:
         print("PyTorch not found, installing default version...")
@@ -127,6 +144,18 @@ def install_torch_sparse():
                 print("Successfully installed torch-sparse with alternative version")
                 return True
     
+    # Try installing with --no-build-isolation
+    print("Trying torch-sparse installation with --no-build-isolation...")
+    if run_command(f"pip install torch-sparse -f {wheel_url} --no-build-isolation"):
+        print("Successfully installed torch-sparse with --no-build-isolation")
+        return True
+    
+    # Try installing from source with --no-build-isolation
+    print("Trying torch-sparse installation from source...")
+    if run_command("pip install torch-sparse --no-build-isolation"):
+        print("Successfully installed torch-sparse from source")
+        return True
+    
     return False
 
 def install_torch_scatter():
@@ -177,6 +206,18 @@ def install_torch_scatter():
                 print("Successfully installed torch-scatter with alternative version")
                 return True
     
+    # Try installing with --no-build-isolation
+    print("Trying torch-scatter installation with --no-build-isolation...")
+    if run_command(f"pip install torch-scatter -f {wheel_url} --no-build-isolation"):
+        print("Successfully installed torch-scatter with --no-build-isolation")
+        return True
+    
+    # Try installing from source with --no-build-isolation
+    print("Trying torch-scatter installation from source...")
+    if run_command("pip install torch-scatter --no-build-isolation"):
+        print("Successfully installed torch-scatter from source")
+        return True
+    
     return False
 
 def install_nvdiffrast():