cpu only

Author: cmanzo

docker/Dockerfile

@@ -21,9 +21,15 @@ COPY .. .
 # ---------- extra Python deps -----------------------------------------------------
 # CPU-only PyTorch (wheels from CPU index)
 RUN pip install --no-cache-dir \
-        torch==2.7.0 torchvision==0.22.0 torchaudio==2.7.0 \
+        torch==2.7.0 torchvision==0.22.0\
         --index-url https://download.pytorch.org/whl/cpu
 
+# CPU-only PyG (and its C/C++ extensions)
+RUN pip install --no-cache-dir \
+        torch-geometric \
+        -f https://data.pyg.org/whl/torch-2.7.0+cpu.html
+
+
 COPY docker/requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 

docker/requirements.txt

@@ -18,7 +18,6 @@ scikit-image==0.25.2
 scikit-learn==1.6.1
 spacy==3.8.6
 tifffile==2025.5.10
-torch_geometric==2.6.1
 torchmetrics==1.6.1
 tqdm==4.67.1
 transformers==4.51.3