Fix PyTorch pip version tags

Dockerfile

@@ -25,14 +25,14 @@ RUN apt-get update && apt-get install -y \
 USER user
 
 # 1. Install PyTorch FIRST (Required for PyG C++ extensions)
-RUN pip install --no-cache-dir --user torch==2.2.0+cpu torchvision==0.17.0+cpu -f https://download.pytorch.org/whl/cpu
+RUN pip install --no-cache-dir --user torch==2.10.0 torchvision==0.25.0 --extra-index-url https://download.pytorch.org/whl/cpu
 
 # 2. Install PyTorch Geometric dependencies (Sparse/Scatter) natively from PyG wheel index
-RUN pip install --no-cache-dir --user torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.2.0+cpu.html
+RUN pip install --no-cache-dir --user torch-scatter torch-sparse -f https://data.pyg.org/whl/torch-2.10.0+cpu.html
 
 # 3. Copy requirements and install the rest (It will skip torch and torch-scatter since they are already installed)
 COPY --chown=user:user requirements.txt .
-RUN pip install --no-cache-dir --user -r requirements.txt -f https://data.pyg.org/whl/torch-2.2.0+cpu.html
+RUN pip install --no-cache-dir --user -r requirements.txt -f https://data.pyg.org/whl/torch-2.10.0+cpu.html
 
 # Copy the rest of the application
 COPY --chown=user:user . .

requirements.txt

@@ -11,8 +11,8 @@ joblib
 
 # Neural Networks & Vision
 --extra-index-url https://download.pytorch.org/whl/cpu
-torch==2.2.0+cpu
-torchvision==0.17.0+cpu
+torch==2.10.0
+torchvision==0.25.0
 segmentation-models-pytorch==0.5.0
 torch-geometric==2.7.0
 torch-geometric-temporal==0.56.2