Update Dockerfile

Dockerfile

@@ -5,16 +5,24 @@ RUN apt-get update && \
     apt-get install -y --no-install-recommends \
         build-essential \
         git \
+        wget \
     && rm -rf /var/lib/apt/lists/* && \
     python -m pip install --upgrade pip
 
 WORKDIR /app
 
-# Install PyTorch first (CPU version)
-RUN pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
+# Install specific PyTorch version (CPU)
+RUN pip install --no-cache-dir torch==2.4.0 --index-url https://download.pytorch.org/whl/cpu
 
-# Install PyTorch Geometric dependencies with pre-built wheels
-RUN pip install --no-cache-dir torch-scatter torch-sparse torch-cluster torch-spline-conv -f https://data.pyg.org/whl/torch-2.5.0+cpu.html
+# Install PyTorch Geometric and extensions matching torch 2.4.0 CPU
+RUN pip install --no-cache-dir \
+    torch-scatter \
+    torch-sparse \
+    torch-cluster \
+    torch-spline-conv \
+    -f https://data.pyg.org/whl/torch-2.4.0+cpu.html
+
+RUN pip install --no-cache-dir torch-geometric
 
 # Copy and install remaining requirements
 COPY ./mcp_output/requirements.txt /app/requirements.txt