Update requirements.txt

requirements.txt

@@ -1,17 +1,21 @@
---extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.1.2
-torchvision==0.16.2
-torchaudio==2.1.2
+# --- PyTorch (CPU) ---
+--index-url https://download.pytorch.org/whl/cpu
+torch==2.2.2
+torchvision==0.17.2
+torchaudio==2.2.2
 
--f https://data.pyg.org/whl/torch-2.1.2+cpu.html
-pyg_lib
+# --- PyG core ---
+torch-geometric==2.5.3
+
+# --- PyG extension wheels (match TORCH=2.2.0, CUDA=cpu) ---
+-f https://data.pyg.org/whl/torch-2.2.0+cpu.html
+pyg-lib
 torch-scatter
 torch-sparse
 torch-cluster
 torch-spline-conv
-torch-geometric
 
-# rest of your deps...
+# --- the rest of your deps ---
 numpy>=1.24
 pandas>=2.0
 scikit-learn>=1.3