Upload 3 files

post_install.py

@@ -57,7 +57,9 @@ def install_torch_sparse():
             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)
+            # Uninstall current PyTorch and reinstall with correct version
+            print("Uninstalling current PyTorch...")
+            run_command("pip uninstall torch torchvision torchaudio -y")
             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
@@ -71,10 +73,28 @@ def install_torch_sparse():
     print("Checking PyTorch installation...")
     check_pytorch_cuda()
     
+    # Verify PyTorch version was actually updated
+    try:
+        import torch
+        if not torch.__version__.startswith("2.0.1") or "+cu117" not in torch.__version__:
+            print(f"Warning: PyTorch version is still {torch.__version__}, expected 2.0.1+cu117")
+            print("This may cause compatibility issues with torch-sparse and torch-scatter")
+    except Exception as e:
+        print(f"Error checking PyTorch version: {e}")
+    
     # Now install torch-sparse with the compatible version
     print("Installing torch-sparse with PyTorch 2.0.1...")
     if run_command("pip install torch-sparse -f https://data.pyg.org/whl/torch-2.0.1+cu117.html"):
         print("Successfully installed torch-sparse")
+        
+        # Verify torch-sparse is compatible
+        try:
+            import torch_sparse
+            print("torch-sparse import successful")
+        except Exception as e:
+            print(f"Warning: torch-sparse import failed: {e}")
+            print("This may indicate a version compatibility issue")
+        
         return True
     
     return False
@@ -95,6 +115,15 @@ def install_torch_scatter():
     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"):
         print("Successfully installed torch-scatter")
+        
+        # Verify torch-scatter is compatible
+        try:
+            import torch_scatter
+            print("torch-scatter import successful")
+        except Exception as e:
+            print(f"Warning: torch-scatter import failed: {e}")
+            print("This may indicate a version compatibility issue")
+        
         return True
     
     return False
@@ -286,6 +315,10 @@ def main():
         import torch
         print(f"✓ PyTorch {torch.__version__} - CUDA: {torch.cuda.is_available()}")
         
+        # Check if PyTorch version is compatible
+        if not torch.__version__.startswith("2.0.1") or "+cu117" not in torch.__version__:
+            print(f"⚠ Warning: PyTorch version {torch.__version__} may not be compatible with installed extensions")
+        
         import torch_sparse
         print("✓ torch-sparse")
         
@@ -303,9 +336,11 @@ def main():
         
     except ImportError as e:
         print(f"✗ Import error: {e}")
+        print("This may indicate a version compatibility issue between PyTorch and its extensions")
         sys.exit(1)
     except Exception as e:
         print(f"✗ Verification error: {e}")
+        print("This may indicate a version compatibility issue between PyTorch and its extensions")
         sys.exit(1)
 
 if __name__ == "__main__":