Upload post_install.py

post_install.py

@@ -71,22 +71,20 @@ def install_torch_sparse():
         
         if pytorch_base.startswith("2.7"):
             print("PyTorch 2.7.x detected - this is very recent and may not have compatible wheels")
-            print("Downgrading to PyTorch 2.0.1+cu117 for better compatibility...")
+            print("Attempting to work with current PyTorch version and disable problematic extensions...")
             
-            # Uninstall current PyTorch
-            print("Uninstalling current PyTorch...")
-            run_command("pip uninstall torch torchvision torchaudio -y")
+            # Keep current PyTorch version but disable torch-sparse and torch-scatter
+            PYTORCH_VERSION = pytorch_base
+            CUDA_VERSION = cuda_version
+            
+            # Force disable torch-sparse and torch-scatter since they have ABI issues
+            print("Disabling torch-sparse and torch-scatter due to ABI compatibility issues...")
+            disable_torch_sparse()
             
-            # Install compatible PyTorch version
-            print("Installing PyTorch 2.0.1+cu117...")
-            if run_command("pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu117"):
-                PYTORCH_VERSION = "2.0.1"
-                CUDA_VERSION = "cu117"
-                print("Successfully installed PyTorch 2.0.1+cu117")
-            else:
-                print("Failed to install PyTorch 2.0.1+cu117, keeping current version")
-                PYTORCH_VERSION = pytorch_base
-                CUDA_VERSION = cuda_version
+            # Install torchvision for current PyTorch version
+            print("Installing torchvision for current PyTorch version...")
+            if not run_command("pip install torchvision"):
+                print("Warning: Failed to install torchvision")
         else:
             # Store version info for later use
             PYTORCH_VERSION = pytorch_base
@@ -515,6 +513,17 @@ def main():
                     print("✗ torch-sparse (not available)")
             except:
                 print("✗ torch-sparse (not available)")
+        except Exception as e:
+            # Check if torch-sparse was disabled due to ABI issues
+            try:
+                with open("NeuralJacobianFields/PoissonSystem.py", 'r') as f:
+                    content = f.read()
+                if "USE_TORCH_SPARSE = False" in content:
+                    print("✓ torch-sparse (disabled due to ABI issues, using built-in PyTorch sparse)")
+                else:
+                    print(f"✗ torch-sparse import failed: {e}")
+            except:
+                print(f"✗ torch-sparse import failed: {e}")
         
         # Check if torch-scatter is available
         try:
@@ -522,6 +531,8 @@ def main():
             print("✓ torch-scatter")
         except ImportError:
             print("⚠ torch-scatter (not available, may not be critical)")
+        except Exception as e:
+            print(f"⚠ torch-scatter import failed: {e} (may not be critical)")
         
         import nvdiffrast
         print("✓ nvdiffrast")